ccibeekeoc42 commited on
Commit
d2439d3
·
verified ·
1 Parent(s): 7ec764e

Updated the files for downloading the tokenizer using the requests library

Browse files
Files changed (1) hide show
  1. app.py +20 -2
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
- !wget https://huggingface.co/novateur/WavTokenizer-medium-speech-75token/resolve/main/wavtokenizer_mediumdata_frame75_3s_nq1_code4096_dim512_kmeans200_attn.yaml
23
- !wget https://huggingface.co/novateur/WavTokenizer-large-speech-75token/resolve/main/wavtokenizer_large_speech_320_24k.ckpt
 
 
 
 
 
 
 
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