Update main.py

main.py

@@ -38,8 +38,8 @@ def get_epsilons():
 # -------------------------------
 # Google Drive model download (irregular-flat)
 # -------------------------------
-MODEL_PATH_IRREGULAR = "/app/model_irregular_flat.pth"
-DRIVE_FILE_ID = "15vi4zPhCs3aBnGepVnXFOqQjxdK1jpnA"  # Your new model
+MODEL_PATH_IRREGULAR = "/tmp/model_irregular_flat.pth"  # <-- changed here
+DRIVE_FILE_ID = "15vi4zPhCs3aBnGepVnXFOqQjxdK1jpnA"
 
 def download_irregular_model():
     if not os.path.exists(MODEL_PATH_IRREGULAR):
@@ -64,13 +64,15 @@ def download_irregular_model():
 
 # Download the irregular model at startup
 download_irregular_model()
+
 # -------------------------------
-# Debug check: confirm model exists
+# Debug check
 # -------------------------------
 if os.path.exists(MODEL_PATH_IRREGULAR):
-    print(" Irregular-flat model is ready at", MODEL_PATH_IRREGULAR)
+    print("Irregular-flat model is ready at", MODEL_PATH_IRREGULAR)
 else:
-    print("Irregular-flat model NOT found. Something went wrong!")
+    print("Irregular-flat model NOT found! Something went wrong!")
+
 # -------------------------------
 # Detectron2 model setup
 # -------------------------------