app.py

import gradio as gr
from transformers import pipeline

# def greet(name):
#     return "Hello " + name + "!!"

# demo = gr.Interface(fn=greet, inputs="text", outputs="text")
# demo.launch()

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


def puntuacion_sentimiento(texto):
    resultado = clasificador(texto)

    print(resultado)

    etiqueta = resultado[0]["label"]

    if (etiqueta == "POS"):
        respuesta = "Tu frase es positiva"
    elif etiqueta == "NEG":
        respuesta = "Tu frase es negativa"
    else:
        respuesta = "Ni fu ni fa"

    return respuesta


demo = gr.Interface(
    fn=puntuacion_sentimiento,
    inputs=gr.Textbox(),
    outputs=gr.Textbox(),
    title="Analizador de Sentimientos",
    description="Analiza el sentimiento de una frase utilizando un modelo preentrenado.",
    article="Escribe una frase y obtén su puntuación de sentimiento.",
    # theme="soft",
    submit_btn="Analizar",
    fill_width=True,
)
demo.launch()