Spaces:
Running
Running
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() | |