Bernardo Damiani commited on
Commit
bc22ea4
·
1 Parent(s): e1ed64f

feat: adiciona novos modelos

Browse files
Files changed (1) hide show
  1. app.py +4 -9
app.py CHANGED
@@ -7,18 +7,16 @@ import torch
7
 
8
  st.set_page_config(page_title="Análise de Sentimento com Transformers", layout="wide")
9
 
10
- # Título
11
  st.title("🔍 Análise de Sentimentos com Transformers")
12
  st.markdown("Insira textos e escolha um modelo de linguagem para realizar a classificação de sentimentos.")
13
 
14
- # Modelos disponíveis
15
  modelos_disponiveis = {
16
- "FinBert (Financeiro - PT-BR)": "turing-usp/FinBertPTBR",
17
- "BERT Base PT-BR (NeuralMind)": "neuralmind/bert-base-portuguese-cased",
18
- "Multilingual BERT": "nlptown/bert-base-multilingual-uncased-sentiment"
 
19
  }
20
 
21
- # Sidebar com seleção de modelo
22
  modelo_escolhido = st.sidebar.selectbox("Escolha o modelo:", list(modelos_disponiveis.keys()))
23
  model_name = modelos_disponiveis[modelo_escolhido]
24
 
@@ -40,7 +38,6 @@ def avaliar_sentimento(texto):
40
  sentimento_previsto = max(resultado, key=resultado.get)
41
  return {"sentimento": sentimento_previsto, "scores": resultado}
42
 
43
- # Entrada de textos
44
  textos = st.text_area("Digite os textos (um por linha):", height=200)
45
 
46
  if st.button("🔍 Analisar Sentimentos"):
@@ -61,11 +58,9 @@ if st.button("🔍 Analisar Sentimentos"):
61
  df_resultados = pd.DataFrame(resultados)
62
  st.success("Análise concluída!")
63
 
64
- # Mostrar tabela
65
  st.subheader("📋 Resultados")
66
  st.dataframe(df_resultados, use_container_width=True)
67
 
68
- # Gráfico
69
  st.subheader("📊 Distribuição de Sentimentos")
70
  fig, ax = plt.subplots()
71
  df_resultados["Sentimento"].value_counts().plot(kind='bar', ax=ax, rot=0)
 
7
 
8
  st.set_page_config(page_title="Análise de Sentimento com Transformers", layout="wide")
9
 
 
10
  st.title("🔍 Análise de Sentimentos com Transformers")
11
  st.markdown("Insira textos e escolha um modelo de linguagem para realizar a classificação de sentimentos.")
12
 
 
13
  modelos_disponiveis = {
14
+ "FinBertPTBR (turing-usp) ": "turing-usp/FinBertPTBR",
15
+ "Multilingual Uncased BERT (nlptown)": "nlptown/bert-base-multilingual-uncased-sentiment",
16
+ "Multilingual Cased BERT (lxyuan)": "lxyuan/distilbert-base-multilingual-cased-sentiments-student",
17
+ "Multilingual Sentiment Analysis (tabularisai)" : "tabularisai/multilingual-sentiment-analysis"
18
  }
19
 
 
20
  modelo_escolhido = st.sidebar.selectbox("Escolha o modelo:", list(modelos_disponiveis.keys()))
21
  model_name = modelos_disponiveis[modelo_escolhido]
22
 
 
38
  sentimento_previsto = max(resultado, key=resultado.get)
39
  return {"sentimento": sentimento_previsto, "scores": resultado}
40
 
 
41
  textos = st.text_area("Digite os textos (um por linha):", height=200)
42
 
43
  if st.button("🔍 Analisar Sentimentos"):
 
58
  df_resultados = pd.DataFrame(resultados)
59
  st.success("Análise concluída!")
60
 
 
61
  st.subheader("📋 Resultados")
62
  st.dataframe(df_resultados, use_container_width=True)
63
 
 
64
  st.subheader("📊 Distribuição de Sentimentos")
65
  fig, ax = plt.subplots()
66
  df_resultados["Sentimento"].value_counts().plot(kind='bar', ax=ax, rot=0)