prueba / app.py
ang16's picture
Finalización codigo
1800e31
raw
history blame contribute delete
597 Bytes
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 ()