| 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() | |