Update

.gitignore

@@ -0,0 +1,2 @@
+main.py
+video.mp4

app.py

@@ -3,12 +3,15 @@ from ultralytics import YOLO
 
 def run(source):
     global model
-    return model.predict(source=frame, conf=.5, iou=.5, stream=False, save=True, show=False)
-
+    res = model(source)
+    res_plotted = res[0].plot()
+    return res_plotted
+    
 model = YOLO("yolov8n-nckh2023.pt") # Select YOLO model
-iface = gr.Interface(
-    fn=run,
-    inputs=gr.inputs.Image(label="Upload a image or video", type="filepath"),
+
+gr.Interface(
+    predict,
+    inputs=gr.inputs.Image(label="Upload hot dog candidate", type="filepath"),
     outputs=gr.Image(shape=[256,256], elem_id="output_image").style(width=256, height=256),
-    title="Motorcyclist, helmet, and license plate detection")
-iface.launch()
+    title="Motorcyclist, helmet, and license plate detection",
+).launch()