Update app.py

app.py

@@ -1,6 +1,7 @@
 from ultralytics import YOLO
 from PIL import Image
 import gradio as gr
+import numpy as np
 def function(image):
   model = YOLO('best_yolo_2.pt')
   image = np.array(image)