import gradio as gr
from fastai.vision.all import *
from huggingface_hub import from_pretrained_fastai
modelo = from_pretrained_fastai("pagaliv/futurama-character-classifier")

def clasificar_imagen(img):
    pred, _, prob = modelo.predict(img)
    return {modelo.dls.vocab[i]: float(prob[i]) for i in range(len(prob))}

# Interfaz
iface = gr.Interface(
    fn=clasificar_imagen,
    inputs=gr.Image(type="pil"),
    outputs=gr.Label(num_top_classes=3),
    examples=["BENDER.jpg","BENDER2.jpg", "FRY.jpg","FRY2.jpg","LEELA.jpg","LEELA2.jpg"],  # Asegúrate de subir estos archivos
    title="Clasificador de Futurama"
)

iface.launch()