add func parameter

app.py

@@ -65,7 +65,7 @@ def summarize_transcription(transcription, model, gemini_prompt):
         return f"Error summarizing transcription: {str(e)}"
     
 @spaces.GPU(duration=120)
-def process_audio(audio_file, language):
+def process_audio(audio_file, language, whisper_model):
     print("Starting transcription...")
 
     if device == 0:
@@ -74,8 +74,8 @@ def process_audio(audio_file, language):
         pipe = load_pipeline("openai/whisper-tiny")
 
     with open(audio_file, "rb") as f:
-            inputs = f.read()
-            
+        inputs = f.read()
+
     inputs = ffmpeg_read(inputs, pipe.feature_extractor.sampling_rate)
     inputs = {"array": inputs, "sampling_rate": pipe.feature_extractor.sampling_rate}
 
@@ -112,7 +112,7 @@ def transcribe(youtube_url, audio_file, whisper_model, gemini_api_key, gemini_pr
             print(f"Audio file read: {audio_file}")
         
         progress(0.4, desc="Starting transcription")
-        transcription = process_audio(audio_file, language)
+        transcription = process_audio(audio_file, language, whisper_model)
         
         progress(0.6, desc="Cleaning up")
         # Delete the audio file after transcription