Update app.py

app.py

@@ -1,11 +1,18 @@
+import spaces
 import gradio as gr
 import os
 import torch
 from pyannote.audio import Pipeline
 
+
+
+
 # Configuration
 HF_TOKEN = os.environ.get('HF_TOKEN')
 
+
+
+
 # Pyannote Diarization
 diarization_pipeline = None
 try:
@@ -22,6 +29,10 @@ try:
 except Exception as e:
     print(f"Pyannote Error: {e}")
 
+
+
+
+@spaces.GPU
 def diarize_audio(audio_path, min_speakers=1, max_speakers=5):
     if not diarization_pipeline:
         return {"error": "Diarization not available. Check HF_TOKEN."}
@@ -34,6 +45,9 @@ def diarize_audio(audio_path, min_speakers=1, max_speakers=5):
     except Exception as e:
         return {"error": str(e)}
 
+
+
+
 with gr.Blocks(title="STTR - Speaker Diarization") as demo:
     gr.Markdown("# STTR - Speaker Diarization")
     gr.Markdown("### Identify who speaks when (pyannote 3.1)")
@@ -46,5 +60,8 @@ with gr.Blocks(title="STTR - Speaker Diarization") as demo:
     output = gr.JSON(label="Speaker Segments")
     btn.click(diarize_audio, [audio_in, min_spk, max_spk], output, api_name="/diarize")
 
+
+
+
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=int(os.environ.get("PORT", 7860)))