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