Spaces:
Sleeping
Sleeping
File size: 483 Bytes
b6ca290 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
import cv2
import numpy as np
import gradio as gr
from ultralytics import YOLO
def predict(path:str):
model = YOLO("yolov8s.yaml")
model = YOLO("best-carvana.pt")
imagen = cv2.imread(path)
results = model.predict(source=path)
for r in results:
return r.plot()
gr.Interface(fn=predict,
inputs=gr.components.Image(type="filepath", label="Input"),
outputs=gr.components.Image(type="numpy", label="Output")).launch(debug=False) |