Spaces:
Running
Running
Fix tqdm
Browse files- translate.py +1 -1
translate.py
CHANGED
@@ -174,7 +174,7 @@ def main(
|
|
174 |
|
175 |
print("\n".join(tgt_text), file=output_file)
|
176 |
|
177 |
-
pbar.update(len(tgt_text))
|
178 |
|
179 |
inference()
|
180 |
print(f"Translation done.\n")
|
|
|
174 |
|
175 |
print("\n".join(tgt_text), file=output_file)
|
176 |
|
177 |
+
pbar.update(len(tgt_text) // gen_kwargs["num_return_sequences"])
|
178 |
|
179 |
inference()
|
180 |
print(f"Translation done.\n")
|