import cv2
import numpy as np
import gradio as gr

from ultralytics import YOLO

from pickle import TRUE
def predict(path:str):
  model = YOLO('yolov8s.yaml')
#aquí se pone el nombre del documento  
  model = YOLO('best.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)