Fix EncoderASR.transcribe_batch() error

app.py

@@ -22,12 +22,16 @@ def transcribe(audio):
         # Load audio
         waveform, sample_rate = torchaudio.load(audio)
 
-        # Ensure correct sample rate (16kHz expected)
+        # Ensure correct sample rate (16kHz expected by the model)
         if sample_rate != 16000:
-            waveform = torchaudio.transforms.Resample(orig_freq=sample_rate, new_freq=16000)(waveform)
+            resampler = torchaudio.transforms.Resample(orig_freq=sample_rate, new_freq=16000)
+            waveform = resampler(waveform)
+
+        # Compute waveform length in seconds relative to batch size
+        wav_lens = torch.tensor([waveform.shape[1] / 16000], dtype=torch.float32)
 
         # Transcribe
-        transcription = asr_model.transcribe_batch(waveform)
+        transcription = asr_model.transcribe_batch(waveform, wav_lens)
         return transcription[0]
 
     except Exception as e: