ArunNyp7 commited on
Commit
b9b132b
·
verified ·
1 Parent(s): e897ace

Update app.py

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