Spaces:
Sleeping
Sleeping
# Importa librerias | |
from transformers import pipeline | |
import gradio as gr | |
# Inicializa Modelo de Clasificaci贸n de Sentimientos | |
model_name = 'pysentimiento/robertuito-sentiment-analysis' | |
classifier = pipeline("text-classification", model=model_name) | |
def classify_text(text): | |
""" | |
Clasifica un texto como positivo, negativo o neutro utilizando un clasificador. | |
Args: | |
text (str): El texto a clasificar. | |
Returns: | |
str: La clasificaci贸n del texto, que puede ser "Positivo", "Negativo" o "Neutro". | |
""" | |
result = classifier(text)[0]['label'] | |
if result == "POS": | |
return "Positivo" | |
elif result == "NEG": | |
return "Negativo" | |
else: | |
return "Neutro" | |
# Crea Interfaz Gradio | |
input_text = gr.inputs.Textbox(label="Texto a clasificar") | |
output_text = gr.outputs.Textbox(label="Sentimiento") | |
gr.Interface(fn=classify_text, inputs=input_text, outputs=output_text, examples=[ | |
['Estoy feliz 馃 de mostrarles un toolkit para An谩lisis de Sentimientos y otras tareas de SocialNLP'], | |
['Espero que no lo odien.'], | |
['Lo odiamos.'] | |
]).launch() | |