Update app.py
Browse files
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=
|
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 |
|