Update app.py

app.py

@@ -2,6 +2,8 @@ import gradio as gr
 from address_extractor import AddressExtractor
 import tempfile
 import os
+import librosa
+import soundfile as sf
 
 # Instantiate your AddressExtractor class
 address_extractor = AddressExtractor()
@@ -32,6 +34,8 @@ def extract_from_audio(audio_file):
     #     tmp_file_path = tmp_file.name
 
     try:
+        audio, sr = librosa.load(audio_file, sr=16000)
+        sf.write(audio_file, audio, 16000)
         # segments = address_extractor.whisper_model.transcribe(tmp_file_path)
         segments = address_extractor.whisper_model.transcribe(audio_file)
         input_text = " ".join([seg.text.strip() for seg in segments])