import cv2
import random
import numpy as np
import gradio as gr

from ultralytics import YOLO


def predict(path):
    model = YOLO("yolov8s.yaml")
    model = YOLO("best.pt")
    results = model.predict(source=path)

    for r in results:
        return r.plot(show_conf=True, line_width=None, font_size=None, pil=False)


gr.Interface(title='Generación 16 - Módulo 4',
             fn=predict,
             inputs=gr.components.Image(type="filepath", label="Input"),
             outputs=gr.components.Image(type="numpy", label="Output")).launch()