sachinsen1295 commited on
Commit
3bfa8d4
·
verified ·
1 Parent(s): a4c60ec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ import datetime
15
  # Load models
16
  embedding_model = PretrainedSpeakerEmbedding(
17
  "speechbrain/spkrec-ecapa-voxceleb",
18
- device=torch.device("cpu") # Use "cuda" if a GPU is available
19
  )
20
  audio_processor = Audio()
21
 
 
15
  # Load models
16
  embedding_model = PretrainedSpeakerEmbedding(
17
  "speechbrain/spkrec-ecapa-voxceleb",
18
+ device=torch.device("cuda") # Use "cuda" if a GPU is available
19
  )
20
  audio_processor = Audio()
21