MiPrimerEspacio / app.py
Alberto Caro
Ejercicio difusion
e0247e5
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="text", outputs="text")
demo.launch()