Update main.py
Browse files
main.py
CHANGED
@@ -59,7 +59,7 @@ def load_assets():
|
|
59 |
try:
|
60 |
print(" Downloading EfficientNet classification model...")
|
61 |
class_model_path = hf_hub_download(
|
62 |
-
repo_id="sheikh987
|
63 |
filename="efficientnet_augmented_best.pth",
|
64 |
cache_dir=cache_dir
|
65 |
)
|
|
|
59 |
try:
|
60 |
print(" Downloading EfficientNet classification model...")
|
61 |
class_model_path = hf_hub_download(
|
62 |
+
repo_id="sheikh987/efficientnet-isic",
|
63 |
filename="efficientnet_augmented_best.pth",
|
64 |
cache_dir=cache_dir
|
65 |
)
|