Update README.md

README.md

@@ -42,7 +42,7 @@ model.eval().cuda()
 
  
 wav, sr = sf.read("test.wav")   
-wav_tensor = torch.from_numpy(wav).float().unsqueeze(0)  
+wav_tensor = torch.from_numpy(wav).float().unsqueeze(0)  # Shape: (1, T)
 
  
 with torch.no_grad():
@@ -51,7 +51,7 @@ with torch.no_grad():
     vq_code = model.encode_code(input_waveform=wav_tensor)
     print("Code:", vq_code )  
 
-    recon_wav = model.decode_code(vq_code).cpu()
+    recon_wav = model.decode_code(vq_code).cpu()       # Shape: (1, 1, T')
 
  
 sf.write("reconstructed.wav", recon_wav[0, 0, :].numpy(), sr)