Update app.py

app.py

@@ -15,7 +15,8 @@ import datetime
 # Load models
 embedding_model = PretrainedSpeakerEmbedding(
     "speechbrain/spkrec-ecapa-voxceleb",
-    device=torch.device("cpu")  # Use "cuda" if a GPU is available
+    device=torch.device("cuda"),  # Use "cuda" if a GPU is available
+    weights_only=True  # Add this parameter to avoid the warning
 )
 audio_processor = Audio()
 
@@ -84,7 +85,7 @@ def diarize(audio_file, num_speakers, model_size="medium"):
 interface = gr.Interface(
     fn=diarize,
     inputs=[
-        gr.Audio(type="file", label="Upload Audio File"),  # Removed 'source' argument
+        gr.Audio(type="filepath", label="Upload Audio File"),  # Use 'filepath' here
         gr.Number(label="Number of Speakers", value=2, precision=0),
         gr.Radio(["tiny", "base", "small", "medium", "large"], label="Model Size", value="medium")
     ],