File size: 929 Bytes
41bc236 d885584 41bc236 d885584 41bc236 d885584 0460e8d d885584 0460e8d d22bf09 0460e8d 41bc236 |
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 29 30 31 |
import gradio as gr
from transformers import pipeline
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="Esta es una muestra para probar <strong>modelos de IA</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()
|