Noursine commited on
Commit
b37ad86
·
verified ·
1 Parent(s): 3b55fa7

Update detectron_infer.py

Browse files
Files changed (1) hide show
  1. 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
- os.environ["GDOWN_CACHE_DIR"] = "/tmp/gdown"
18
- os.makedirs("/tmp/gdown", exist_ok=True)
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...")