Update app.py

app.py

@@ -7,7 +7,6 @@ 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"]]
@@ -16,11 +15,10 @@ image_examples = [["DurianMangosteen1.jpg"], ["DurianMangosteen2.jpg"]]
 def load_model(repo_id):
     download_dir = snapshot_download(repo_id) # download the model from the Hugging Face Hub
     print(download_dir)
-    #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
+    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(best_yolo_model, task='detect')# load the model
     return detection_model
 
 # Predict the image
@@ -43,7 +41,7 @@ def predict(pilimg):
         return out_pilimg
 
 REPO_ID = "ITI107-2024S2/8035531F"  # The repo ID of the model
-#detection_model = load_model(REPO_ID)
+detection_model = load_model(REPO_ID)
 
 title = "Detect Durian and Mangosteen (King and Queen of Fruits) In The Image"
 interface = gr.Interface(