File size: 558 Bytes
6fa2c2a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
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() |