Update README.md
Browse files
README.md
CHANGED
|
@@ -37,6 +37,7 @@ Please notice that we encourage you to read our tutorials and learn more about
|
|
| 37 |
|
| 38 |
### Perform speech-to-speech translation (S2ST) with S2UT model and the Vocoder
|
| 39 |
```python
|
|
|
|
| 40 |
import torchaudio
|
| 41 |
from speechbrain.pretrained import EncoderDecoderS2UT
|
| 42 |
from speechbrain.pretrained import UnitHIFIGAN
|
|
@@ -47,6 +48,7 @@ hifi_gan_unit = UnitHIFIGAN.from_hparams(source="speechbrain/tts-hifigan-unit-hu
|
|
| 47 |
|
| 48 |
# Running the S2UT model
|
| 49 |
codes = s2ut.translate_file("speechbrain/s2st-transformer-fr-en-hubert-l6-k100-cvss/example-fr.wav")
|
|
|
|
| 50 |
|
| 51 |
# Running Vocoder (units-to-waveform)
|
| 52 |
waveforms = hifi_gan_unit.decode_unit(codes)
|
|
|
|
| 37 |
|
| 38 |
### Perform speech-to-speech translation (S2ST) with S2UT model and the Vocoder
|
| 39 |
```python
|
| 40 |
+
import torch
|
| 41 |
import torchaudio
|
| 42 |
from speechbrain.pretrained import EncoderDecoderS2UT
|
| 43 |
from speechbrain.pretrained import UnitHIFIGAN
|
|
|
|
| 48 |
|
| 49 |
# Running the S2UT model
|
| 50 |
codes = s2ut.translate_file("speechbrain/s2st-transformer-fr-en-hubert-l6-k100-cvss/example-fr.wav")
|
| 51 |
+
codes = torch.IntTensor(codes)
|
| 52 |
|
| 53 |
# Running Vocoder (units-to-waveform)
|
| 54 |
waveforms = hifi_gan_unit.decode_unit(codes)
|