Update app.py

app.py

@@ -1,7 +1,11 @@
 import gradio as gr
 
-def greet(name):
-    return "Hello " + name + "!!"
+from ultralytics import YOLO
 
-demo = gr.Interface(fn=greet, inputs="image", outputs="text")
+model = YOLO("best.pt")
+
+def greet(detection):
+    return model(detection)
+
+demo = gr.Interface(fn=greet, inputs="image", outputs="image")
 demo.launch()