Update app.py

app.py

@@ -1,3 +1,27 @@
 import gradio as gr
+from pyannote.audio import Pipeline
 
-gr.load("models/pyannote/speaker-diarization").launch()
+# Load the diarization pipeline
+pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization")
+
+# Function to handle diarization
+def diarize_audio(file_path):
+    diarization = pipeline(file_path)
+
+    # Create a result to show the speaker segments
+    result = []
+    for segment, _, speaker in diarization.itertracks(yield_label=True):
+        result.append(f"Speaker {speaker}: from {segment.start:.1f} to {segment.end:.1f}")
+    return "\n".join(result)
+
+# Create Gradio interface
+interface = gr.Interface(
+    fn=diarize_audio,
+    inputs=gr.Audio(source="upload", type="filepath"),
+    outputs="text",
+    title="Speaker Diarization",
+    description="Upload an audio file to perform speaker diarization."
+)
+
+# Launch the Gradio app
+interface.launch()