anilkumar5590 commited on
Commit
f3bf76d
·
verified ·
1 Parent(s): 3c6ac45

Update backendapi.py

Browse files
Files changed (1) hide show
  1. backendapi.py +1 -1
backendapi.py CHANGED
@@ -54,7 +54,7 @@ def load_model_from_huggingface():
54
  print("Downloading model from Hugging Face Hub...")
55
 
56
  model_path = hf_hub_download(
57
- repo_id="your-huggingface-username/your-model-repo",
58
  filename="model.keras", # Adjust the filename based on your model
59
  cache_dir=cache_dir # Use the custom directory
60
  )
 
54
  print("Downloading model from Hugging Face Hub...")
55
 
56
  model_path = hf_hub_download(
57
+ repo_id="anilkumar5590/image-classification-viit",
58
  filename="model.keras", # Adjust the filename based on your model
59
  cache_dir=cache_dir # Use the custom directory
60
  )