tclopess's picture
Update app.py
25d615d verified
"""
from transformers import pipeline
import gradio as gr
import torch
pipe = pipeline('sentiment-analysis')
def get_sentiment(input_text):
return pipe(input_text)
app_examples = [
['Não estou em minha melhor versão'],
['Relaxado como nunca'],
['Tentando relaxar, porém preocupado ']
]
inputs = [
gr.Textbox(value=app_examples[0][0]),
]
iface = gr.Interface(fn=get_sentiment,
inputs=inputs,
outputs=["text"],
title='Sentiment Analysis',
description='Obtenha o sentimento do texto digitado 😄|😠',
examples= app_examples,
theme="gradio/monochrome"
)
iface.launch()
"""
from transformers import pipeline
import gradio as gr
import torch
from pysentimiento import create_analyzer
analyzer = create_analyzer(task="sentiment", lang="pt")
def get_sentiment(input_text):
prever = analyzer.predict([input_text])
return prever
app_examples = [
['A comunicação ineficaz entre as equipes dificultou o cumprimento dos prazos'],
['Só sei que nada sei'],
['Tentando relaxar, porém preocupado ']
]
inputs = [
gr.Textbox(value=app_examples[0][0]),
]
iface = gr.Interface(fn=get_sentiment,
inputs=inputs,
outputs=["text"],
title='Sentiment Analysis',
description='Obtenha o sentimento do texto digitado 😄😶😠',
examples= app_examples,
theme="gradio/monochrome"
)
iface.launch()