Update app.py

app.py

@@ -38,8 +38,8 @@ except Exception as e:
 
 # --- Download and Load Classification Model (EfficientNet) ---
 try:
-    CLASS_REPO_ID = "sheikh987/efficientnet-isic-classifier"
-    CLASS_MODEL_FILENAME = "efficientnet_isic_classifier_best.pth"
+    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)