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:0" if torch.cuda.is_available() else "cpu")
+    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     diarization_pipeline.to(device)
 except Exception as e:
     print(f"Error initializing diarization pipeline: {e}")
@@ -36,7 +36,7 @@ MODEL_NAME = "openai/whisper-medium"
 FILE_LIMIT_MB = 1000
 YT_LENGTH_LIMIT_S = 3600
 
-device = "cuda:0" if torch.cuda.is_available() else "cpu"
+device = 0 if torch.cuda.is_available() else "cpu"
 
 pipe = pipeline(
     task="automatic-speech-recognition",
@@ -49,6 +49,7 @@ pipe = pipeline(
 
 
 
+
 def associate_speakers_with_timestamps(transcription_result, diarization, tolerance=0.1, min_segment_duration=0.5):
     word_segments = transcription_result['chunks']
     diarization_segments = list(diarization.itertracks(yield_label=True))