import gradio as gr
import torch
import numpy as np
from PIL import Image

model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt', _verbose=False)

def gradio_wrapper(img):
    global model
    results = model(img)
    return results.render()[0]

image = gr.Image(source="webcam", streaming=True)



demo = gr.Interface(
    gradio_wrapper,
    image,
    'image',
    live=True,
    title="CiclopeIA",
    description="Aplicación basada en el proyecto CiclopeIA de Saturdays AI. Identifica el valor de los billetes de Euro."
)

demo.launch()