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)