Spaces:
Running
Running
Update inference.py
Browse files- inference.py +1 -1
inference.py
CHANGED
@@ -18,7 +18,7 @@ class InferencePipeline():
|
|
18 |
self.hifigan = torch.hub.load("bshall/hifigan:main", "hifigan_hubert_soft", trust_repo=True, map_location=torch.device('cpu'))
|
19 |
|
20 |
# load target speaker embedding
|
21 |
-
self.trg_spk_emb = np.load('heidi.npy')#'p225_007_mic1.npy')
|
22 |
self.trg_spk_emb = torch.from_numpy(self.trg_spk_emb)
|
23 |
self.trg_spk_emb = self.trg_spk_emb.unsqueeze(0)#.cuda()
|
24 |
|
|
|
18 |
self.hifigan = torch.hub.load("bshall/hifigan:main", "hifigan_hubert_soft", trust_repo=True, map_location=torch.device('cpu'))
|
19 |
|
20 |
# load target speaker embedding
|
21 |
+
self.trg_spk_emb = np.load('p226_007_mic1.npy')#'heidi.npy')#'p225_007_mic1.npy')
|
22 |
self.trg_spk_emb = torch.from_numpy(self.trg_spk_emb)
|
23 |
self.trg_spk_emb = self.trg_spk_emb.unsqueeze(0)#.cuda()
|
24 |
|