8513669W commited on
Commit
f296a50
·
verified ·
1 Parent(s): e532bfe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ model_path = "C:/Users/KimBok/Downloads/best_int8_openvino_model"
10
  def load_model(repo_id):
11
  download_dir = snapshot_download(repo_id)
12
  print(download_dir)
13
- path = os.path.join(download_dir, "best_int8_openvino_model1")
14
  print(path)
15
  detection_model = YOLO(path, task='detect')
16
  return detection_model
 
10
  def load_model(repo_id):
11
  download_dir = snapshot_download(repo_id)
12
  print(download_dir)
13
+ path = os.path.join(download_dir, "best_int8_openvino_model")
14
  print(path)
15
  detection_model = YOLO(path, task='detect')
16
  return detection_model