Dharini Baskaran commited on
Commit
1f3b592
·
1 Parent(s): 7524599

updated the model weight path

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -15,9 +15,10 @@ BASE_DIR = os.path.dirname(os.path.abspath(__file__))
15
  UPLOAD_DIR = "/tmp/uploads/"
16
  JSON_DIR = "/tmp/results/"
17
  OUTPUT_DIR = "/tmp/output/"
 
18
  MODEL_DIR = os.path.join(BASE_DIR, "rcnn_model", "scripts")
19
  logo_path = os.path.join(BASE_DIR, "public", "logo.png")
20
- model_path = os.path.join(OUTPUT_DIR, "model_final_2.pth")
21
 
22
  # Google Drive model
23
  GOOGLE_DRIVE_FILE_ID = "1ptdr6d_jglk_78e4d5s1jUO04XWS1zo"
@@ -27,6 +28,7 @@ GDRIVE_URL = f"https://drive.google.com/uc?id={GOOGLE_DRIVE_FILE_ID}"
27
  os.makedirs(UPLOAD_DIR, exist_ok=True)
28
  os.makedirs(JSON_DIR, exist_ok=True)
29
  os.makedirs(OUTPUT_DIR, exist_ok=True)
 
30
 
31
  # Download model if missing
32
  if not os.path.exists(model_path):
 
15
  UPLOAD_DIR = "/tmp/uploads/"
16
  JSON_DIR = "/tmp/results/"
17
  OUTPUT_DIR = "/tmp/output/"
18
+ OUTPUT_DIR2= "/tmp/output2/"
19
  MODEL_DIR = os.path.join(BASE_DIR, "rcnn_model", "scripts")
20
  logo_path = os.path.join(BASE_DIR, "public", "logo.png")
21
+ model_path = os.path.join(OUTPUT_DIR2, "model_final_2.pth")
22
 
23
  # Google Drive model
24
  GOOGLE_DRIVE_FILE_ID = "1ptdr6d_jglk_78e4d5s1jUO04XWS1zo"
 
28
  os.makedirs(UPLOAD_DIR, exist_ok=True)
29
  os.makedirs(JSON_DIR, exist_ok=True)
30
  os.makedirs(OUTPUT_DIR, exist_ok=True)
31
+ os.makedirs(OUTPUT_DIR2, exist_ok=True)
32
 
33
  # Download model if missing
34
  if not os.path.exists(model_path):