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