Update app.py

app.py

@@ -7,18 +7,11 @@ Original file is located at
     https://colab.research.google.com/drive/1GzjDFYPEtsFsBFnhi3x3B0vWyCE-Dtpb
 """
 
-import os
+from ultralytics import YOLO
 from PIL import Image
 import gradio as gr
 from huggingface_hub import snapshot_download
-
-# Ensure necessary packages are installed
-try:
-    from ultralytics import YOLO
-except ImportError:
-    import subprocess
-    subprocess.check_call(["pip", "install", "ultralytics"])
-    from ultralytics import YOLO
+import os
 
 model_path = "/Users/markk/Downloads/best_int8_openvino_model"
 
@@ -30,27 +23,24 @@ def load_model(repo_id):
     detection_model = YOLO(path, task='detect')
     return detection_model
 
+
 def predict(pilimg):
     source = pilimg
     result = detection_model.predict(source, conf=0.5, iou=0.6)
     img_bgr = result[0].plot()
     out_pilimg = Image.fromarray(img_bgr[..., ::-1])  # RGB-order PIL image
-
+    
     return out_pilimg
 
-def predict_video(video_path):
-    result = detection_model.predict(video_path, conf=0.5, iou=0.6)
-    return result[0].save("output_video.mp4")  # Save processed video
-
 REPO_ID = "Lesterchia174/Monkey_Durian"
 detection_model = load_model(REPO_ID)
 
+video_url = "https://github.com/lesterchia1/Monkey_Durian/blob/main/Monkey_Durian.mp4"
+
 gr.Interface(
     fn=predict,
     inputs=gr.Image(type="pil"),
-    outputs=gr.Image(type="pil")
+    outputs=gr.Image(type="pil"),
+    title="Monkey Durian Detector",
+    description=f"[Click here to watch the video]({video_url})"
 ).launch(share=True)
-
-# Process the new video file
-video_path = "Monkey_Durian.mp4"
-predict_video(video_path)