Spaces:
Runtime error
Runtime error
from transformers import pipeline # <- Simplifica o uso de modelos pre-treinados | |
def sentiment_analysis(input_user): | |
''' | |
Essa função recolhe a mensagem do | |
user e retorna o sentimento (positivo/negativo). | |
''' | |
# Definir tarefa de classificação | |
pipe = pipeline("sentiment-analysis") # <- Modelo pode ser personalizado | |
response = pipe(input_user) # <- Mensagem para ser avaliada | |
for r in response: | |
return (r['label']) | |
# Importar Gradio | |
import gradio as gr | |
# Criação da interface | |
interface = gr.Interface(fn=sentiment_analysis, # <- Função que retorna classificação do texto | |
inputs=[gr.Textbox(label="Mensagem", placeholder="Mensagem")], # <- Campo de texto | |
outputs=gr.Textbox(label="Classificação"), # <- Resposta | |
title="Sentiment-Analysis" | |
) | |
# Run | |
interface.launch(share=True) |