Update app.py

app.py

@@ -28,7 +28,6 @@ def merge_entities(entities):
             current = next_entity
     merged.append(current)
     return merged
-
 def transcribe_audio(audio_path):
     """
     Transcribe a local audio file using the Whisper pipeline, log timing, and save transcription to a file.
@@ -40,6 +39,9 @@ def transcribe_audio(audio_path):
         # Ensure audio is mono and resampled to 16kHz
         audio, sr = librosa.load(audio_path, sr=16000, mono=True)
 
+        # Set the pad token for batching
+        whisper_pipeline_agri.tokenizer.pad_token_id = whisper_pipeline_agri.model.config.eos_token_id
+
         # Perform transcription
         transcription = whisper_pipeline_agri(audio, batch_size=8)["text"]
 
@@ -51,6 +53,7 @@ def transcribe_audio(audio_path):
     except Exception as e:
         return f"Error processing audio: {e}"
 
+
 def translate_text_to_english(text):
     """
     Translate text into English using GoogleTranslator.