jhauret commited on
Commit
4781d40
1 Parent(s): 8cb4d2b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -59,7 +59,7 @@ Detailed instructions for reproducing the experiments are available on the [jhau
59
 
60
  ```python
61
  import torch, torchaudio
62
- from vibravox import EBENGenerator
63
  from datasets import load_dataset
64
 
65
  model = EBENGenerator.from_pretrained("Cnam-LMSSC/EBEN_forehead_accelerometer")
@@ -68,7 +68,7 @@ test_dataset = load_dataset("Cnam-LMSSC/vibravox", "speech_clean", split="test",
68
  audio_48kHz = torch.Tensor(next(iter(test_dataset))["audio.forehead_accelerometer"]["array"])
69
  audio_16kHz = torchaudio.functional.resample(audio_48kHz, orig_freq=48_000, new_freq=16_000)
70
 
71
- cut_audio_16kHz = model.cut_to_valid_length(audio_16kHz)
72
  enhanced_audio_16kHz = model(cut_audio_16kHz)
73
  ```
74
 
 
59
 
60
  ```python
61
  import torch, torchaudio
62
+ from vibravox.torch_modules.dnn.eben_generator import EBENGenerator
63
  from datasets import load_dataset
64
 
65
  model = EBENGenerator.from_pretrained("Cnam-LMSSC/EBEN_forehead_accelerometer")
 
68
  audio_48kHz = torch.Tensor(next(iter(test_dataset))["audio.forehead_accelerometer"]["array"])
69
  audio_16kHz = torchaudio.functional.resample(audio_48kHz, orig_freq=48_000, new_freq=16_000)
70
 
71
+ cut_audio_16kHz = model.cut_to_valid_length(audio_16kHz[None, None, :])
72
  enhanced_audio_16kHz = model(cut_audio_16kHz)
73
  ```
74