Update
Browse files
app.py
CHANGED
@@ -6,5 +6,9 @@ def run(source):
|
|
6 |
return model.predict(source=frame, conf=.5, iou=.5, stream=False, save=True, show=False)
|
7 |
|
8 |
model = YOLO("yolov8n-nckh2023.pt") # Select YOLO model
|
9 |
-
iface = gr.Interface(
|
|
|
|
|
|
|
|
|
10 |
iface.launch()
|
|
|
6 |
return model.predict(source=frame, conf=.5, iou=.5, stream=False, save=True, show=False)
|
7 |
|
8 |
model = YOLO("yolov8n-nckh2023.pt") # Select YOLO model
|
9 |
+
iface = gr.Interface(
|
10 |
+
fn=run,
|
11 |
+
inputs=gr.inputs.Image(label="Upload a image or video", type="filepath"),
|
12 |
+
outputs=gr.outputs.Image(label="Image output"),
|
13 |
+
title="Motorcyclist, helmet, and license plate detection")
|
14 |
iface.launch()
|