Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import os
 import torch
 import io
+import wavio
 from pyannote.audio import Pipeline
 from pyannote.audio import Audio
 from pyannote.core import Segment
@@ -11,12 +12,12 @@ pipeline = Pipeline.from_pretrained(
   use_auth_token=os.environ['api'])
 
 def process_audio(audio):
-    # Extract the audio data from the tuple
-    audio_data, _ = audio
+    # Extract the audio data and sample rate from the tuple
+    audio_data, sample_rate = audio
     
     # Save the uploaded audio file to a temporary location
-    with open("temp.wav", "wb") as f:
-        f.write(audio_data)
+    with wavio.open("temp.wav", "w", rate=sample_rate, channels=1, sampwidth=2) as wav:
+        wav.write(audio_data)
     
     # Use the diarization pipeline to process the audio
     diarization = pipeline("temp.wav")