File size: 567 Bytes
99c36ed
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
# Use a pipeline as a high-level helper
from transformers import pipeline
import gradio as gr

pipe = pipeline("text-classification", model="Genius1237/xlm-roberta-large-tydip")

def puntuacion_sentimientos(texto):
    resultado = pipe(texto)
    print(resultado)
    etiqueta = resultado[0]["label"]
    if(etiqueta == "polite" ):
        respuesta = "Tu frase es educada"
    elif etiqueta == "impolite":
        respuesta = "Tu frase es grosera"
    return respuesta
    
demo = gr.Interface(fn=puntuacion_sentimientos, inputs="text", outputs="text")
demo.launch()