Update app.py

app.py

@@ -13,6 +13,7 @@ from typing import Optional, Tuple
 import torch
 import numpy as np
 import torchaudio
+from cached_path import cached_path
 
 # Add MuCodec to path
 sys.path.insert(0, "./MuCodec")
@@ -50,7 +51,8 @@ print("Language model loaded!")
 print("Loading MuCodec decoder...")
 mucodec_dir = "./MuCodec"
 ckpt_path = os.path.join(mucodec_dir, "ckpt/mucodec.pt")
-audioldm_path = os.path.join(mucodec_dir, "tools/audioldm_48k.pth")
+#audioldm_path = os.path.join(mucodec_dir, "tools/audioldm_48k.pth")
+audioldm_path = str(cached_path("hf://haoheliu/audioldm_48k/audioldm_48k.pth"))
 config_path = os.path.join(mucodec_dir, "configs/models/transformer2D.json")
 
 # Load VAE and STFT