Update app.py

app.py

@@ -22,7 +22,7 @@ from pyannote.audio import Audio
 from pyannote.core import Segment
 import wave
 import contextlib
-from sklearn.cluster import SpectralClustering
+from sklearn.cluster import KMeans
 
 import numpy as np
 import json
@@ -227,7 +227,7 @@ def Transcribe_V2(num_speakers, speaker_names, audio="temp_audio.wav"):
         return embedding_model(waveform[None])
 
     def add_speaker_labels(segments, embeddings, num_speakers):
-        clustering = SpectralClustering(num_speakers).fit(embeddings)
+        clustering = KMeans(num_speakers).fit(embeddings)
         labels = clustering.labels_
         for i in range(len(segments)):
             segments[i]["speaker"] = 'SPEAKER ' + str(labels[i] + 1)