Update
Browse files
app.py
CHANGED
@@ -5,6 +5,6 @@ def run(source):
|
|
5 |
global model
|
6 |
return model.predict(source=frame, conf=.5, iou=.5, stream=False, save=True, show=False)
|
7 |
|
8 |
-
model = YOLO("
|
9 |
-
iface = gr.Interface(fn=run, inputs="
|
10 |
iface.launch()
|
|
|
5 |
global model
|
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(fn=run, inputs="image", outputs="image")
|
10 |
iface.launch()
|