File size: 738 Bytes
48027df 28ac00e 48027df 28ac00e 48027df 28ac00e 48027df |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
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() |