Spaces:
Runtime error
Runtime error
rvian
commited on
Commit
·
3740b02
1
Parent(s):
64a313a
ajuste carregar modelo
Browse files
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 |
-
|
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)
|