Update app.py

app.py

@@ -4,8 +4,20 @@ import torchaudio
 from einops import rearrange
 from stable_audio_tools import get_pretrained_model
 from stable_audio_tools.inference.generation import generate_diffusion_cond
-# Download model
-model, model_config = get_pretrained_model("stabilityai/stable-audio-open-1.0")
+from huggingface_hub import cached_download, hf_hub_url
+from transformers import AutoModelForAudioClassification
+import os
+
+token = os.get.environ("TOKEN")
+model_name = "stabilityai/stable-audio-open-1.0"
+model_config_url = hf_hub_url(repo_id=model_name, revision="main", filename="model_config.json")
+model_config = cached_download(model_config_url, use_auth_token=token)
+
+model = AutoModelForAudioClassification.from_pretrained(
+    model_name, 
+    cache_dir=None,
+    use_auth_token=token
+)
 sample_rate = model_config["sample_rate"]
 sample_size = model_config["sample_size"]