Hugging Face's logo

Spaces:
chompionsawelo
/
whisper_transcribe
Runtime error

App Files Community
1
chompionsawelo commited on
Commit
18e34db
·
1 Parent(s): bdec318

change to cuda

Browse files
Files changed (1) hide show
  1. diarization.py +1 -1
diarization.py CHANGED
@@ -7,7 +7,7 @@ import json
7
  hugging_face_token = os.environ["HUGGING_FACE_TOKEN"]
8
  pipeline = Pipeline.from_pretrained(
9
  'pyannote/speaker-diarization', use_auth_token=hugging_face_token)
10
- device = torch.device("cpu")
11
  pipeline.to(device)
12
 
13
 
 
7
  hugging_face_token = os.environ["HUGGING_FACE_TOKEN"]
8
  pipeline = Pipeline.from_pretrained(
9
  'pyannote/speaker-diarization', use_auth_token=hugging_face_token)
10
+ device = torch.device("cuda")
11
  pipeline.to(device)
12
 
13