Update detectron_infer.py

detectron_infer.py

@@ -14,14 +14,23 @@ DRIVE_FILE_ID = "1TNECsdcZ82tLOFxWTQMdLPGlvM3c5JWp"
 
 def download_model():
     if not os.path.exists(MODEL_PATH):
-        # Use a writable cache directory under /tmp instead of /
-        gdown_cache = "/tmp/gdown"
-        os.environ["GDOWN_CACHE_DIR"] = gdown_cache
-        os.makedirs(gdown_cache, exist_ok=True)
-
         url = f"https://drive.google.com/uc?id={DRIVE_FILE_ID}"
+
+        # Create a writable temporary directory for gdown
+        tmp_dir = "/tmp/gdown"
+        os.makedirs(tmp_dir, exist_ok=True)
+
+        # Set GDOWN cache path to a safe location
+        os.environ["GDOWN_CACHE_DIR"] = tmp_dir
+
         print("Downloading Detectron2 model...")
-        gdown.download(url, MODEL_PATH, quiet=False)
+        gdown.download(
+            url,
+            MODEL_PATH,
+            quiet=False,
+            fuzzy=True,
+            use_cookies=False  # Important for Hugging Face
+        )
         print("Download complete.")
 
 download_model()