Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -52,6 +52,7 @@ For more demos, please goto [https://voila.maitrix.org](https://voila.maitrix.or
|
|
| 52 |
default_ref_emb_mask_list = pickle.load(open(default_ref_file, "rb"))
|
| 53 |
million_voice_ref_emb_mask_list = pickle.load(open(million_voice_ref_file, "rb"))
|
| 54 |
|
|
|
|
| 55 |
def get_ref_embs(ref_audio):
|
| 56 |
wav, sr = torchaudio.load(ref_audio)
|
| 57 |
ref_embs = spkr_model(wav, sr).cpu()
|
|
|
|
| 52 |
default_ref_emb_mask_list = pickle.load(open(default_ref_file, "rb"))
|
| 53 |
million_voice_ref_emb_mask_list = pickle.load(open(million_voice_ref_file, "rb"))
|
| 54 |
|
| 55 |
+
@spaces.GPU
|
| 56 |
def get_ref_embs(ref_audio):
|
| 57 |
wav, sr = torchaudio.load(ref_audio)
|
| 58 |
ref_embs = spkr_model(wav, sr).cpu()
|