Skip to content

Commit c2bc010

Browse files
authored
Update README.md
1 parent 2c761ff commit c2bc010

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

README.md

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ pip install pythaiasr
1919
```
2020

2121
**For Wav2Vec2 with language model:**
22-
if you want to use wannaphong/wav2vec2-large-xlsr-53-th-cv8-* model, you needs to install by the step.
22+
if you want to use wannaphong/wav2vec2-large-xlsr-53-th-cv8-* model with language model, you needs to install by the step.
2323

2424
```sh
2525
pip install pythaiasr[lm]
@@ -37,17 +37,19 @@ print(asr(file))
3737
### API
3838

3939
```python
40-
asr(file: str, model: str = "airesearch/wav2vec2-large-xlsr-53-th")
40+
asr(file: str, model: str = _model_name, lm: bool=False, device: str=None)
4141
```
4242

4343
- file: path of sound file
4444
- model: The ASR model
45+
- lm: Use language model (except *airesearch/wav2vec2-large-xlsr-53-th* model)
46+
- device: device
4547
- return: thai text from ASR
4648

4749
**Options for model**
4850
- *airesearch/wav2vec2-large-xlsr-53-th* (default) - AI RESEARCH - PyThaiNLP model
49-
- *wannaphong/wav2vec2-large-xlsr-53-th-cv8-newmm* - Thai Wav2Vec2 with CommonVoice V8 (newmm tokenizer) + language model
50-
- *wannaphong/wav2vec2-large-xlsr-53-th-cv8-deepcut* - Thai Wav2Vec2 with CommonVoice V8 (deepcut tokenizer) + language model
51+
- *wannaphong/wav2vec2-large-xlsr-53-th-cv8-newmm* - Thai Wav2Vec2 with CommonVoice V8 (newmm tokenizer)
52+
- *wannaphong/wav2vec2-large-xlsr-53-th-cv8-deepcut* - Thai Wav2Vec2 with CommonVoice V8 (deepcut tokenizer)
5153

5254
You can read about models from the list:
5355

0 commit comments

Comments
 (0)