Spaces:

rvian
/

traduzir_dataset_en_ptbr

Runtime error

rvian commited on Sep 12, 2023

Commit

c034f10

1 Parent(s): c18735b

aa

Files changed (1) hide show

app.py CHANGED Viewed

@@ -79,7 +79,8 @@ if dataset is not None:
 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)
-    resultado()

 if st.button("Carregar modelo"):
     _modelo, _tokenizador = carregar_modelo(modelos_opcao)
+    print(f"{_modelo.name_or_path} carregado")
+    if st.button("Traduzir dataset") and _modelo is not None:
+        traduzir_dataset(dataset, _modelo, _tokenizador)
+        resultado()