Spaces:

sheikh987
/

Sc-classifier

Sleeping

sheikh987 commited on Jul 13

Commit

fb9a6d7

verified ·

1 Parent(s): 95a831b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -38,7 +38,7 @@ except Exception as e:
 # --- Download and Load Classification Model (EfficientNet) ---
 try:
-    CLASS_REPO_ID = "sheikh987/efficientnet_augmented_best"
     CLASS_MODEL_FILENAME = "efficientnet_augmented_best.pth"
     print(f"--> Downloading classification model from: {CLASS_REPO_ID}")
     class_model_path = hf_hub_download(repo_id=CLASS_REPO_ID, filename=CLASS_MODEL_FILENAME)

 # --- Download and Load Classification Model (EfficientNet) ---
 try:
+    CLASS_REPO_ID = "sheikh987/efficientnet-isic"
     CLASS_MODEL_FILENAME = "efficientnet_augmented_best.pth"
     print(f"--> Downloading classification model from: {CLASS_REPO_ID}")
     class_model_path = hf_hub_download(repo_id=CLASS_REPO_ID, filename=CLASS_MODEL_FILENAME)