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