Update README.md

README.md

@@ -9,6 +9,11 @@ tags:
 datasets:
 - Cnam-LMSSC/vibravox
 ---
+
+<p align="center">
+  <img src="https://cdn-uploads.huggingface.co/production/uploads/65302a613ecbe51d6a6ddcec/zhB1fh-c0pjlj-Tr4Vpmr.png" style="object-fit:contain; width:280px; height:280px;" >
+</p>
+
 # Model Card 
 
 - **Developed by:** [Cnam-LMSSC](https://huggingface.co/Cnam-LMSSC)
@@ -39,7 +44,11 @@ import torch, torchaudio
 from vibravox import EBENGenerator
 from datasets import load_dataset
 
-audio_16kHz, _ = torch.load("path_to_audio")
+model = EBENGenerator.from_pretrained("Cnam-LMSSC/EBEN_soft_in_ear_microphone")
+test_dataset = load_dataset("Cnam-LMSSC/vibravox", "speech_clean", split="test", streaming=True)
+
+audio_48kHz = torch.Tensor(next(iter(test_dataset))["audio.soft_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)
 enhanced_audio_16kHz = model(cut_audio_16kHz)