wedf
Browse files
app.py
CHANGED
|
@@ -35,9 +35,8 @@ locale.getpreferredencoding = getpreferredencoding
|
|
| 35 |
embedding_model = PretrainedSpeakerEmbedding(
|
| 36 |
"speechbrain/spkrec-ecapa-voxceleb",
|
| 37 |
device=torch.device("cpu"))
|
| 38 |
-
model = whisper.load_model(model_size)
|
| 39 |
audio = Audio()
|
| 40 |
-
torch.set_default_dtype(torch.float32)
|
| 41 |
|
| 42 |
def segment_embedding(segment,duration,path):
|
| 43 |
start = segment["start"]
|
|
|
|
| 35 |
embedding_model = PretrainedSpeakerEmbedding(
|
| 36 |
"speechbrain/spkrec-ecapa-voxceleb",
|
| 37 |
device=torch.device("cpu"))
|
| 38 |
+
model = whisper.load_model(model_size).to_device(device)
|
| 39 |
audio = Audio()
|
|
|
|
| 40 |
|
| 41 |
def segment_embedding(segment,duration,path):
|
| 42 |
start = segment["start"]
|