ArunNyp7 commited on
Commit
69c260e
·
verified ·
1 Parent(s): f836189

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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
@@ -27,7 +27,7 @@ def predict(pilimg):
27
  return out_pilimg
28
 
29
 
30
- REPO_ID = "ITI107-2024S2/1944757V"
31
  detection_model = load_model(REPO_ID)
32
 
33
  gr.Interface(fn=predict,
 
9
  def load_model(repo_id):
10
  download_dir = snapshot_download(repo_id)
11
  print(download_dir)
12
+ path = os.path.join(download_dir)
13
  print(path)
14
  detection_model = YOLO(path, task='detect')
15
  return detection_model
 
27
  return out_pilimg
28
 
29
 
30
+ REPO_ID = "ArunNyp7/ObjectDetectionModel1944"
31
  detection_model = load_model(REPO_ID)
32
 
33
  gr.Interface(fn=predict,