Dharini Baskaran
commited on
Commit
·
1f3b592
1
Parent(s):
7524599
updated the model weight path
Browse files
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(
|
| 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):
|