Update README.md
Browse files
README.md
CHANGED
|
@@ -41,10 +41,21 @@ pip install sidlingvo
|
|
| 41 |
Example usage:
|
| 42 |
|
| 43 |
```Python
|
| 44 |
-
import
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 45 |
|
| 46 |
wav_file = "your_wav_file.wav"
|
| 47 |
-
runner = wav_to_lang.WavToLangRunner(
|
|
|
|
|
|
|
|
|
|
| 48 |
top_lang, _ = runner.wav_to_lang(wav_file)
|
| 49 |
print("Predicted language:", top_lang)
|
| 50 |
```
|
|
|
|
| 41 |
Example usage:
|
| 42 |
|
| 43 |
```Python
|
| 44 |
+
import os
|
| 45 |
+
from sidlingvo import wav_to_lang
|
| 46 |
+
from huggingface_hub import hf_hub_download
|
| 47 |
+
|
| 48 |
+
repo_id = "tflite-hub/conformer-lang-id"
|
| 49 |
+
model_path = "models"
|
| 50 |
+
hf_hub_download(repo_id=repo_id, filename="vad_short_model.tflite", local_dir=model_path)
|
| 51 |
+
hf_hub_download(repo_id=repo_id, filename="vad_short_mean_stddev.csv", local_dir=model_path)
|
| 52 |
+
hf_hub_download(repo_id=repo_id, filename="conformer_langid_medium.tflite", local_dir=model_path)
|
| 53 |
|
| 54 |
wav_file = "your_wav_file.wav"
|
| 55 |
+
runner = wav_to_lang.WavToLangRunner(
|
| 56 |
+
vad_model_file=os.path.join(model_path, "vad_short_model.tflite"),
|
| 57 |
+
vad_mean_stddev_file=os.path.join(model_path, "vad_short_mean_stddev.csv"),
|
| 58 |
+
langid_model_file=os.path.join(model_path, "conformer_langid_medium.tflite"))
|
| 59 |
top_lang, _ = runner.wav_to_lang(wav_file)
|
| 60 |
print("Predicted language:", top_lang)
|
| 61 |
```
|