DiffusersTransformers / app_prueba.py
iespsurmreqmer's picture
Commit sin el token
43a3eff
raw
history blame contribute delete
958 Bytes
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 muy positiva"
elif etiqueta == "NEG":
respuesta = "Tu frase muy negativa"
else:
respuesta = "ni fu ni fa"
return respuesta
demo = gr.Interface(
fn=puntuacion_sentimientos,
inputs=gr.Textbox(),
outputs=gr.Textbox(),
title="Ejemplo Sentimientos de las frases",
description="Esta es nuestra interfaz para probar <strong>modelos de UA</strong>",
article="En este modelo, escribe una frase y pulsa en **Comprobar** para ver si tiene sentimientos positivos, negativos o neutros",
submit_btn="comprobar",
fill_width=True
)
demo.launch(theme=gr.themes.Soft())