Update app.py

app.py

@@ -134,6 +134,7 @@ import gradio as gr
 import torch
 from PIL import Image
 import numpy as np
+from ultralytics import YOLO
 
 # Load the YOLO model
 MODEL_URL= 'https://huggingface.co/ayoubsa/yolo_model/resolve/main/best.pt'