Update app.py

app.py

@@ -1,9 +1,17 @@
+import subprocess
+import sys
+import os
 import torch
 import gradio as gr
 from PIL import Image
-import os
 import zipfile
-from ultralytics import YOLO
+
+# Install ultralytics if not already installed
+try:
+    from ultralytics import YOLO
+except ImportError:
+    subprocess.check_call([sys.executable, "-m", "pip", "install", "ultralytics"])
+    from ultralytics import YOLO
 
 # Load the YOLOv8 model (ensure the path is correct)
 model = YOLO('best-3.pt')  # Path to your trained YOLOv8 model