Update app.py

app.py

@@ -9,7 +9,7 @@ import os
 def load_model(repo_id):
     download_dir = snapshot_download(repo_id)
     print(download_dir)
-    path  = os.path.join(download_dir, "best_int8_openvino_model")
+    path  = os.path.join(download_dir, "best_openvino_model")
     print(path)
     detection_model = YOLO(path, task='detect')
     return detection_model
@@ -27,7 +27,7 @@ def predict(pilimg):
     return out_pilimg
 
 
-REPO_ID = "khengkok/balloon_yolov8"
+REPO_ID = "wooalex/dogs_yolov8"
 detection_model = load_model(REPO_ID)
 
 gr.Interface(fn=predict,