Azzan Dwi Riski commited on
Commit
df8e6a2
·
1 Parent(s): 12c9666

add models

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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-iamge-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)
 
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)