Update app.py

app.py

@@ -8,11 +8,12 @@ import os
 import numpy as np
 
 def translate_speech_to_speech(input_audio):
-    if input_audio.ndim == 2:
-        input_audio = np.mean(input_audio, axis=1)
-      # Save the input audio to a temporary file
+    input_audio, sample_rate = input_tuple
+    
+    # Save the input audio to a temporary file
     input_file = "input_audio.wav"
-    sf.write(input_file, input_audio, 16000)
+    sf.write(input_file, input_audio, sample_rate)  # use the sample rate from Gradio
+
     
     # Language detection and translation code from the first code snippet
     model = whisper.load_model("base")