| 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 positiva" | |
| elif etiqueta == "NEG": | |
| respuesta = "Tu grase es negativa" | |
| else: | |
| respuesta = "Ni fu ni fa" | |
| return respuesta | |
| demo = gr.Interface(fn=puntuacion_sentimientos, | |
| inputs="text", | |
| outputs="text", | |
| description="Esta es nuestra interfaz para probar <strong>modelos de IA</strong>") | |
| demo.launch() |