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)
|