PyThaiASR v1.0.1

- Fixed bug import

Author: wannaphong

README.md

@@ -47,7 +47,7 @@ asr(file: str, show_pad: bool = False, model: str = "airesearch/wav2vec2-large-x
 
 **Options for model**
 - *airesearch/wav2vec2-large-xlsr-53-th* (default) - AI RESEARCH - PyThaiNLP model
-- **wannaphong/wav2vec2-large-xlsr-53-th-cv8-newmm* - Thai Wav2Vec2 with CommonVoice V8 (newmm tokenizer) + language model 
+- *wannaphong/wav2vec2-large-xlsr-53-th-cv8-newmm* - Thai Wav2Vec2 with CommonVoice V8 (newmm tokenizer) + language model 
 - *wannaphong/wav2vec2-large-xlsr-53-th-cv8-deepcut* - Thai Wav2Vec2 with CommonVoice V8 (deepcut tokenizer) + language model 
 
 You can read about models from the list:

pythaiasr/__init__.py

@@ -54,9 +54,9 @@ def __call__(self, file: str, tokenized: bool = False) -> str:
         pred_ids = torch.argmax(logits, dim=-1)[0]
 
         if tokenized:
-            txt = self.processor.decode(pred_ids)
+            txt = self.processor.batch_decode(logits.detach().numpy()).text
         else:
-            txt = self.processor.decode(pred_ids).replace(' ','')
+            txt = self.processor.batch_decode(logits.detach().numpy()).text.replace(' ','')
         return txt
 
 _model_name = "airesearch/wav2vec2-large-xlsr-53-th"

setup.py

@@ -27,7 +27,7 @@ def read(*paths):
 
 setup(
 	name='pythaiasr',
-	version='1.0.0',
+	version='1.0.1',
 	packages=['pythaiasr'],
 	url='https://github.com/pythainlp/pythaiasr',
 	license='Apache Software License 2.0',