Spaces:

TZO777
/

Test_Flowers

Sleeping

TZO777 commited on Jan 14

Commit

68b73c2

verified ·

1 Parent(s): 6449558

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,7 +17,7 @@ def load_model(repo_id):
     # Download the single model file from Hugging Face model repository
-    path = hf_hub_download(repo_id=REPO_ID, filename=MODEL_ID)
     print(path)
     detection_model = YOLO(path, task='detect')

     # Download the single model file from Hugging Face model repository
+    path = hf_hub_download(repo_id=REPO_ID, filename=MODEL_ID, repo_type="space")
     print(path)
     detection_model = YOLO(path, task='detect')