Spaces:
Sleeping
Sleeping
File size: 301 Bytes
0e59ed2 aa749ff 0e59ed2 aa749ff e307b4d 388c2c3 b1fbed0 388c2c3 aa749ff 0e59ed2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
import gradio as gr
from ultralytics import YOLO
model = YOLO("best.pt")
def greet(source):
model = YOLO("best.pt")
results = model.predict(source, imgsz=1080, conf=0.25)
result = results[0]
return result.plot()
demo = gr.Interface(fn=greet, inputs="image", outputs="image")
demo.launch()
|