Update detectron_infer.py
Browse files- detectron_infer.py +2 -3
detectron_infer.py
CHANGED
|
@@ -14,9 +14,8 @@ DRIVE_FILE_ID = "1TNECsdcZ82tLOFxWTQMdLPGlvM3c5JWp"
|
|
| 14 |
|
| 15 |
def download_model():
|
| 16 |
if not os.path.exists(MODEL_PATH):
|
| 17 |
-
|
| 18 |
-
|
| 19 |
-
|
| 20 |
|
| 21 |
url = f"https://drive.google.com/uc?id={DRIVE_FILE_ID}"
|
| 22 |
print("Downloading Detectron2 model...")
|
|
|
|
| 14 |
|
| 15 |
def download_model():
|
| 16 |
if not os.path.exists(MODEL_PATH):
|
| 17 |
+
os.environ["GDOWN_CACHE_DIR"] = "/tmp/gdown"
|
| 18 |
+
os.makedirs("/tmp/gdown", exist_ok=True)
|
|
|
|
| 19 |
|
| 20 |
url = f"https://drive.google.com/uc?id={DRIVE_FILE_ID}"
|
| 21 |
print("Downloading Detectron2 model...")
|