Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -10,7 +10,7 @@ num_classes = 2 # Update with the actual number of classes in your dataset (e.g
|
|
10 |
|
11 |
# Download model from Hugging Face
|
12 |
def download_model():
|
13 |
-
model_path = hf_hub_download(repo_id="
|
14 |
return model_path
|
15 |
|
16 |
# Load the model from Hugging Face
|
|
|
10 |
|
11 |
# Download model from Hugging Face
|
12 |
def download_model():
|
13 |
+
model_path = hf_hub_download(repo_id="jays009/Restnet50", filename="pytorch_model.bin")
|
14 |
return model_path
|
15 |
|
16 |
# Load the model from Hugging Face
|