jays009 commited on
Commit
eff8876
·
verified ·
1 Parent(s): 38d7439

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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="your_huggingface_username/your_model_name", filename="pytorch_model.bin")
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