Hugging Face's logo

Spaces:
breadlicker45
/
WavTokenizer-demo
Build error

App Files Community
breadlicker45 commited on
Commit
e969b0f
·
verified ·
1 Parent(s): f12fa11

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ from decoder.pretrained import WavTokenizer
6
 
7
  # Initialize WavTokenizer
8
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
9
- config_path = "wavtokenizer_smalldata_frame75_3s_nq1_code4096_dim512_kmeans200_attn.yaml"
10
  model_path = "WavTokenizer_small_600_24k_4096.ckpt"
11
 
12
  wavtokenizer = WavTokenizer.from_pretrained0802(config_path, model_path)
 
6
 
7
  # Initialize WavTokenizer
8
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
9
+ config_path = "configs/wavtokenizer_smalldata_frame75_3s_nq1_code4096_dim512_kmeans200_attn.yaml"
10
  model_path = "WavTokenizer_small_600_24k_4096.ckpt"
11
 
12
  wavtokenizer = WavTokenizer.from_pretrained0802(config_path, model_path)