File size: 592 Bytes
632f27a
e0247e5
632f27a
e0247e5
632f27a
e0247e5
 
 
 
 
 
 
 
 
 
 
 
 
632f27a
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
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()