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 = "./configs/wavtokenizer_config.yaml"  # Update with your config path
-model_path = "./wavtokenizer_model.ckpt"  # Update with your model path
+config_path = "wavtokenizer_smalldata_frame75_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)