File size: 1,050 Bytes
2b1a41b 0e7788c 2b1a41b 0e7788c 2b1a41b 0e7788c c8cf501 4c901c4 | 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 es muy positiva"
elif etiqueta == "NEG":
respuesta = "Tu frase es muy negativa"
else:
respuesta = "ni fu ni fa"
return respuesta
demo = gr.Interface(
fn=puntuacion_sentimientos,
inputs=gr.Textbox(placeholder="Escribe tu frase.", label="Frase para analizar"),
outputs=gr.Textbox(label="Sentimiento de la frase"),
title="Ejemplo Sentimientos de las frases",
description="Esta es nuestra interfaz para probar <strong>modelos de IA</strong>",
article="En este modelo, escribe una frase y pulsa en **Comprobar** para ver si tiene setimientos Positivos, Negativos o Neutros",
submit_btn="Comprobar",
fill_width=True
)
demo.launch(theme=gr.themes.Soft()) |