Spaces:

ITI107-2024S2
/

1643512C

Sleeping

miao-zhen commited on Jan 4

Commit

39050d2

1 Parent(s): dba842b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ import os
 def load_model(repo_id):
     download_dir = snapshot_download(repo_id)
     print(download_dir)
-    path  = os.path.join(download_dir, "best_model")
     print(path)
     detection_model = YOLO(path, task='detect')
     return detection_model

 def load_model(repo_id):
     download_dir = snapshot_download(repo_id)
     print(download_dir)
+    path  = os.path.join(download_dir, "best_int8_openvino_model")
     print(path)
     detection_model = YOLO(path, task='detect')
     return detection_model