import gradio as gr
from transformers import pipeline

clasificador = pipeline("sentiment-analysis", model="pysentimiento/robertuito-sentiment-analysis")

def puntuacion_sentimientos(texto):
    resultado = clasificador(texto)
    print(resultado)
    etiqueta = resultado[0]["label"]
    if(etiqueta == "POS"):
        respuesta = "Tu frase es positiva"
    elif etiqueta == "NEG":
        respuesta = "Tu grase es negativa"
    else:
        respuesta = "Ni fu ni fa"
    return respuesta
        

demo = gr.Interface(fn=puntuacion_sentimientos, 
                    inputs="text", 
                    outputs="text",
                    description="Esta es nuestra interfaz para probar <strong>modelos de IA</strong>")
demo.launch()