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