From f86948ebda9d074b3dc7ad9b94eaa73fa6a90902 Mon Sep 17 00:00:00 2001 From: Amar Sattaur Date: Sat, 9 Dec 2023 02:46:19 -0500 Subject: [PATCH] Update main.py --- main.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/main.py b/main.py index 5e9e6f7..9108ac3 100644 --- a/main.py +++ b/main.py @@ -18,10 +18,12 @@ print("e3") +IN_DEV = True MONGO_URI = os.getenv('MONGO_URI') MONGO_DB_NAME = os.getenv('MONGO_DB_NAME') CHROMA_HOST = os.getenv('CHROMA_HOST') print("HOST IS", CHROMA_HOST) +" model_path = "combo_2023-08-02_03:48:00_8.1k_imgs_80_epochs_-1.0000_mse.pth" device = "cpu"