Update app.py

app.py

@@ -7,6 +7,8 @@ import os
 
 model_path = "best_int8_openvino_model"
 
+best_yolo_model = "best.pt" # Trained YOLOv8 detection model
+
 # Example paths for Gradio
 image_examples = [["DurianMangosteen1.jpg"], ["DurianMangosteen2.jpg"]]
 
@@ -17,7 +19,8 @@ def load_model(repo_id):
     #path  = os.path.join(download_dir, "best_int8_openvino_model") # path to the model    
     #print(path)
     #detection_model = YOLO(path, task='detect') # load the model
-    detection_model = YOLO(download_dir, task='detect') # load the model
+    
+    detection_model = YOLO(best_yolo_model, task='detect')# load the model
     return detection_model
 
 # Predict the image