Add model

app.py

@@ -1,7 +1,10 @@
 import gradio as gr
+from ultralytics import YOLO
 
-def greet(name):
-    return "Hello " + name + "!!"
+def run(source):
+    global model
+    return model.predict(source=frame, conf=.5, iou=.5, stream=False, save=True, show=False)
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
+model = YOLO("models/yolov8m-nckh2023.pt") # Select YOLO model
+iface = gr.Interface(fn=run, inputs="text", outputs="image")
 iface.launch()

yolov8n-nckh2023.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e45b0413e7f7eb9987f87f1d27b9187d3fa2941e2c06c8183b09a2ecd03cd451
+size 6249976