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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -15,8 +15,7 @@ import datetime
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
- weights_only=True # Add this parameter to avoid the warning
20
  )
21
  audio_processor = Audio()
22
 
 
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