rvian commited on
Commit
3740b02
·
1 Parent(s): 64a313a

ajuste carregar modelo

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -48,7 +48,7 @@ if dataset is not None:
48
  modelo_selecionado = st.selectbox('Escolha um modelo', modelos_opcao)
49
 
50
  if st.button("Carregar modelo escolhido"):
51
- tokenizer, model = carregar_modelo_e_tokenizador_mbart(modelo_selecionado)
52
  st.write(f"🎰 Modelo {modelo_selecionado} carregado com sucesso! 🔥")
53
 
54
  qtde_linhas_traduzir = st.slider('Quantidade de linhas a serem traduzidas', 1, len(dataset), 50)
 
48
  modelo_selecionado = st.selectbox('Escolha um modelo', modelos_opcao)
49
 
50
  if st.button("Carregar modelo escolhido"):
51
+ model, tokenizer = carregar_modelo_e_tokenizador_mbart(modelo_selecionado)
52
  st.write(f"🎰 Modelo {modelo_selecionado} carregado com sucesso! 🔥")
53
 
54
  qtde_linhas_traduzir = st.slider('Quantidade de linhas a serem traduzidas', 1, len(dataset), 50)