Update README.md

README.md

@@ -59,7 +59,7 @@ Detailed instructions for reproducing the experiments are available on the [jhau
 
 ```python
 import torch, torchaudio
-from vibravox import EBENGenerator
+from vibravox.torch_modules.dnn.eben_generator import EBENGenerator
 from datasets import load_dataset
 
 model = EBENGenerator.from_pretrained("Cnam-LMSSC/EBEN_rigid_in_ear_microphone")
@@ -68,6 +68,6 @@ test_dataset = load_dataset("Cnam-LMSSC/vibravox", "speech_clean", split="test",
 audio_48kHz = torch.Tensor(next(iter(test_dataset))["audio.rigid_in_ear_microphone"]["array"])
 audio_16kHz = torchaudio.functional.resample(audio_48kHz, orig_freq=48_000, new_freq=16_000)
 
-cut_audio_16kHz = model.cut_to_valid_length(audio_16kHz)
+cut_audio_16kHz = model.cut_to_valid_length(audio_16kHz[None, None, :])
 enhanced_audio_16kHz = model(cut_audio_16kHz)
 ```