Update app.py

app.py

@@ -24,7 +24,7 @@ try:
         "pyannote/speaker-diarization-3.1",
         use_auth_token=os.environ["HF_TOKEN"]
     )
-    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+    device = "cuda:0" if torch.cuda.is_available() else "cpu"
     diarization_pipeline.to(device)
 except Exception as e:
     print(f"Error initializing diarization pipeline: {e}")