Update app.py

app.py

@@ -1,3 +1,5 @@
+
+from ultralytics import YOLO
 import gradio as gr
 import numpy as np
 from PIL import Image