Update main.py

main.py

@@ -59,8 +59,8 @@ def load_assets():
     try:
         print("    Downloading EfficientNet classification model...")
         class_model_path = hf_hub_download(
-            repo_id="sheikh987/efficientnet-isic-classifier",
-            filename="efficientnet_isic_classifier_best.pth",
+            repo_id="sheikh987//efficientnet-isic",
+            filename="efficientnet_augmented_best.pth",
             cache_dir=cache_dir
         )
         classification_model = timm.create_model('efficientnet_b3', pretrained=False, num_classes=7).to(DEVICE)