Skip to content

Commit

Permalink
PyThaiASR v1.1.2
Browse files Browse the repository at this point in the history
- remove replace space
  • Loading branch information
wannaphong committed Aug 11, 2022
1 parent 35c953a commit 71fc538
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion pythaiasr/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -53,7 +53,7 @@ def __call__(self, file: str) -> str:
logits = self.model(input_dict.input_values).logits
pred_ids = torch.argmax(logits, dim=-1)[0]
if self.model_name == "airesearch/wav2vec2-large-xlsr-53-th":
txt = self.processor.decode(pred_ids).replace(' ','')
txt = self.processor.decode(pred_ids)
else:
txt = self.processor.batch_decode(logits.detach().numpy()).text[0]
return txt
Expand Down
2 changes: 1 addition & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,7 @@ def read(*paths):

setup(
name='pythaiasr',
version='1.1.1',
version='1.1.2',
packages=['pythaiasr'],
url='https://github.com/pythainlp/pythaiasr',
license='Apache Software License 2.0',
Expand Down

0 comments on commit 71fc538

Please sign in to comment.