CASLL's picture
Update app.py
0335d7b verified
raw
history blame contribute delete
942 Bytes
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)