Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -32,7 +32,7 @@ NUM_CLASSES = len(ID2LABEL)
|
|
32 |
def download_model_from_drive():
|
33 |
# Create a models directory
|
34 |
os.makedirs("models", exist_ok=True)
|
35 |
-
model_path = "models/
|
36 |
|
37 |
if not os.path.exists(model_path):
|
38 |
# Fixed Google Drive URL format for gdown
|
|
|
32 |
def download_model_from_drive():
|
33 |
# Create a models directory
|
34 |
os.makedirs("models", exist_ok=True)
|
35 |
+
model_path = "models/best_model"
|
36 |
|
37 |
if not os.path.exists(model_path):
|
38 |
# Fixed Google Drive URL format for gdown
|