Spaces:

jesseplusplus
/

easy-translate

Running

Iker commited on Jul 1, 2022

Commit

a4abd95

1 Parent(s): 95d0c3a

Fix tqdm

Files changed (1) hide show

translate.py CHANGED Viewed

@@ -174,7 +174,7 @@ def main(
                         print("\n".join(tgt_text), file=output_file)
-                    pbar.update(len(tgt_text))
     inference()
     print(f"Translation done.\n")

                         print("\n".join(tgt_text), file=output_file)
+                    pbar.update(len(tgt_text) // gen_kwargs["num_return_sequences"])
     inference()
     print(f"Translation done.\n")