Spaces:
Sleeping
Sleeping
Updated the files for downloading the tokenizer using the requests library
Browse files
app.py
CHANGED
@@ -15,12 +15,30 @@ import random
|
|
15 |
import uroman as ur
|
16 |
import numpy as np
|
17 |
import torchaudio
|
|
|
|
|
18 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
19 |
from outetts.wav_tokenizer.decoder import WavTokenizer
|
20 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
21 |
|
22 |
-
|
23 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
24 |
|
25 |
|
26 |
from yarngpt.audiotokenizer import AudioTokenizerV2
|
|
|
15 |
import uroman as ur
|
16 |
import numpy as np
|
17 |
import torchaudio
|
18 |
+
import subprocess
|
19 |
+
import requests
|
20 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
21 |
from outetts.wav_tokenizer.decoder import WavTokenizer
|
22 |
|
23 |
+
# Download files using Python's requests library instead of !wget
|
24 |
+
def download_file(url, save_path):
|
25 |
+
response = requests.get(url, stream=True)
|
26 |
+
if response.status_code == 200:
|
27 |
+
with open(save_path, 'wb') as f:
|
28 |
+
f.write(response.content)
|
29 |
+
print(f"Downloaded {save_path}")
|
30 |
+
else:
|
31 |
+
print(f"Failed to download {url}")
|
32 |
|
33 |
+
# Download required files
|
34 |
+
download_file(
|
35 |
+
"https://huggingface.co/novateur/WavTokenizer-medium-speech-75token/resolve/main/wavtokenizer_mediumdata_frame75_3s_nq1_code4096_dim512_kmeans200_attn.yaml",
|
36 |
+
"wavtokenizer_mediumdata_frame75_3s_nq1_code4096_dim512_kmeans200_attn.yaml"
|
37 |
+
)
|
38 |
+
download_file(
|
39 |
+
"https://huggingface.co/novateur/WavTokenizer-large-speech-75token/resolve/main/wavtokenizer_large_speech_320_24k.ckpt",
|
40 |
+
"wavtokenizer_large_speech_320_24k.ckpt"
|
41 |
+
)
|
42 |
|
43 |
|
44 |
from yarngpt.audiotokenizer import AudioTokenizerV2
|