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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -13,9 +13,10 @@ import numpy as np
13
  import datetime
14
 
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
 
 
13
  import datetime
14
 
15
  # Load models
16
+ device = torch.device("cpu") # Explicitly set device to CPU
17
  embedding_model = PretrainedSpeakerEmbedding(
18
+ "speechbrain/spkrec-ecapa-voxceleb",
19
+ device=device # Ensure it uses CPU
20
  )
21
  audio_processor = Audio()
22