Spaces:
Sleeping
Sleeping
Bernardo Damiani
commited on
Commit
·
bc22ea4
1
Parent(s):
e1ed64f
feat: adiciona novos modelos
Browse files
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 |
-
"
|
17 |
-
"
|
18 |
-
"Multilingual BERT": "
|
|
|
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)
|