fix #139

lukas-blecher · Apr 26, 2022 · aa4093f · aa4093f
1 parent 857e6a6
commit aa4093f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pix2tex/dataset/dataset.py b/pix2tex/dataset/dataset.py
@@ -254,7 +254,7 @@ def generate_tokenizer(equations, output, vocab_size):
     tokenizer = Tokenizer(BPE())
     tokenizer.pre_tokenizer = pre_tokenizers.ByteLevel(add_prefix_space=False)
     trainer = BpeTrainer(special_tokens=["[PAD]", "[BOS]", "[EOS]"], vocab_size=vocab_size, show_progress=True)
-    tokenizer.train(trainer, equations)
+    tokenizer.train(equations, trainer)
     tokenizer.save(path=output, pretty=False)
 
 

diff --git a/setup.py b/setup.py
@@ -9,7 +9,7 @@
 
 setuptools.setup(
     name='pix2tex',
-    version='0.0.18',
+    version='0.0.19',
     description='pix2tex: Using a ViT to convert images of equations into LaTeX code.',
     long_description=long_description,
     long_description_content_type='text/markdown',