Spaces:
Sleeping
Sleeping
Azzan Dwi Riski
commited on
Commit
·
df8e6a2
1
Parent(s):
12c9666
add models
Browse files
app.py
CHANGED
@@ -101,7 +101,7 @@ if os.path.exists(image_model_path):
|
|
101 |
print("Image-only model loaded from state_dict successfully!")
|
102 |
else:
|
103 |
print("Image-only model not found locally. Downloading from Hugging Face Hub...")
|
104 |
-
image_model_path = hf_hub_download(repo_id="azzandr/gambling-
|
105 |
image_only_model = models.efficientnet_b3(weights=models.EfficientNet_B3_Weights.DEFAULT)
|
106 |
num_features = image_only_model.classifier[1].in_features
|
107 |
image_only_model.classifier = nn.Linear(num_features, 1)
|
|
|
101 |
print("Image-only model loaded from state_dict successfully!")
|
102 |
else:
|
103 |
print("Image-only model not found locally. Downloading from Hugging Face Hub...")
|
104 |
+
image_model_path = hf_hub_download(repo_id="azzandr/gambling-image-model", filename="best_image_model_Adam_lr0.0001_bs32_state_dict.pt")
|
105 |
image_only_model = models.efficientnet_b3(weights=models.EfficientNet_B3_Weights.DEFAULT)
|
106 |
num_features = image_only_model.classifier[1].in_features
|
107 |
image_only_model.classifier = nn.Linear(num_features, 1)
|