from huggingface_hub import from_pretrained_fastai
import gradio as gr
from fastai.text.all import *

repo_id = "paquinon/entregable3"

learner = from_pretrained_fastai(repo_id)

# Etiquetas legibles para el usuario
label_map = {'0': 'Negativo', '1': 'Positivo'}

def predict(text):
    pred_class, _, _ = learner.predict(text)
    return label_map[str(pred_class)]

gr.Interface(
    fn=predict,
    inputs=gr.Textbox(lines=4, placeholder="Escribe el texto aquí..."),
    outputs=gr.Textbox(label="Predicción"),
    title="Clasificador de opiniones sobre películas",
    description="Introduce un texto para predecir si la opinión es positiva o negativa."
).launch(share=False)