Spaces:
No application file
No application file
File size: 958 Bytes
43a3eff |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
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()) |