agora vai

app.py

@@ -32,7 +32,6 @@ def traduzir_en_pt(text, modelo, tokenizador):
 def carregar_dataset():
     dataset = st.file_uploader("(coluna a ser traduzida deve ser nomeada como 'texto')",type="csv")
     if dataset is not None:
-        st.write('Carregando dataset...')
         return pd.read_csv(dataset)
 
 def mostrar_dataset():
@@ -43,18 +42,18 @@ def mostrar_dataset():
 
 def carregar_modelo(modelos_opcao):
     modelo_selecionado = st.selectbox('Escolha um modelo', modelos_opcao)
-    if st.button("Carregar modelo escolhido"):
+    if modelo_selecionado is not None:
         model, tokenizer = carregar_modelo_e_tokenizador_mbart(modelo_selecionado)
         st.write(f"🎰 Modelo {modelo_selecionado} carregado com sucesso!  🔥")
     return model, tokenizer
 
-def traduzir_dataset(dataset, modelo, tokenizador):
+def traduzir_dataset(dataset, _modelo, _tokenizador):
     qtde_linhas_traduzir = st.slider('Quantidade de linhas a serem traduzidas', 1, len(dataset), 50)
     if st.button(f"Traduzir {qtde_linhas_traduzir} linhas"):
         for i in range(qtde_linhas_traduzir):
             st.write(f'🔡  Traduzindo linha {i+1}...')
             st.write(f'Texto: {dataset.iloc[i]["texto"]}')
-            texto_traduzido= traduzir_en_pt(dataset.iloc[i]["texto"], modelo, tokenizador)
+            texto_traduzido= traduzir_en_pt(dataset.iloc[i]["texto"], _modelo, _tokenizador)
             st.write(f'Tradução: {texto_traduzido}')
 
             # adiciona traducao em nova coluna dataset
@@ -78,10 +77,9 @@ if dataset is not None:
     mostrar_dataset()
     
 
-if dataset is not None:
-    if st.button("Carregar modelo"):
-        modelo, tokenizador = carregar_modelo(modelos_opcao)
+if st.button("Carregar modelo"):
+    _modelo, _tokenizador = carregar_modelo(modelos_opcao)
 
-if st.button("Traduzir dataset") and modelo is not None:
-    traduzir_dataset(dataset, modelo, tokenizador)
+if st.button("Traduzir dataset") and _modelo is not None:
+    traduzir_dataset(dataset, _modelo, _tokenizador)
     resultado()