Update app.py

app.py

@@ -10,11 +10,15 @@ from pyannote.audio import Audio
 from pyannote.core import Segment
 import gradio as gr
 
+
 import yt_dlp as youtube_dl
 from gradio_client import Client
-from pyannote.audio import Pipeline
 from transformers.pipelines.audio_utils import ffmpeg_read
 
+import pyannote.core.json
+
+
+
 HF_TOKEN = "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"
 # set up the diarization pipeline
 diarization_pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization-3.0", use_auth_token=HF_TOKEN)
@@ -30,8 +34,12 @@ def transcribe(audio_path, task="transcribe", group_by_speaker=True, progress=gr
 
     # run diarization while we wait for Whisper JAX
     progress(0, desc="Diarizing...")
-    diarization = diarization_pipeline(audio_path)
-    segments = diarization.for_json()["content"]
+    # diarization = diarization_pipeline(audio_path)
+    # segments = diarization.for_json()["content"]
+
+    # convert diarization to JSON
+    diarization_json = pyannote.core.json.for_json(diarization)
+    segments = diarization_json["content"]
 
     transcription = segments