Update app.py

app.py

@@ -133,6 +133,7 @@ iface.launch()"""
 import gradio as gr
 import torch
 from PIL import Image
+import numpy as np
 
 # Load the YOLO model
 model = torch.hub.load('ultralytics/yolov5', 'custom', path='https://huggingface.co/ayoubsa/yolo_model/blob/main/best.pt')  # Replace with your uploaded model's path