|
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 () |