Update app.py

app.py

@@ -6,8 +6,8 @@ from decoder.pretrained import WavTokenizer
 
 # Initialize WavTokenizer
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-config_path = "wavtokenizer_mediumdata_music_audio_frame75_3s_nq1_code4096_dim512_kmeans200_attn.yaml"
-model_path = "wavtokenizer_medium_music_audio_320_24k.ckpt"
+config_path = "wavtokenizer_smalldata_frame40_3s_nq1_code4096_dim512_kmeans200_attn.yaml"
+model_path = "WavTokenizer_small_600_24k_4096.ckpt"
 
 wavtokenizer = WavTokenizer.from_pretrained0802(config_path, model_path)
 wavtokenizer = wavtokenizer.to(device)