Woziii commited on
Commit
617b530
·
verified ·
1 Parent(s): 65db7e8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -24,7 +24,7 @@ try:
24
  "pyannote/speaker-diarization-3.1",
25
  use_auth_token=os.environ["HF_TOKEN"]
26
  )
27
- device = "cuda:0" if torch.cuda.is_available() else "cpu"
28
  diarization_pipeline.to(device)
29
  except Exception as e:
30
  print(f"Error initializing diarization pipeline: {e}")
 
24
  "pyannote/speaker-diarization-3.1",
25
  use_auth_token=os.environ["HF_TOKEN"]
26
  )
27
+ device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
28
  diarization_pipeline.to(device)
29
  except Exception as e:
30
  print(f"Error initializing diarization pipeline: {e}")