Update app.py

app.py

@@ -20,9 +20,8 @@ def generate_audio(prompt, seconds_total=30, steps=100, cfg_scale=7):
     if not hf_token:
         raise EnvironmentError("HF_TOKEN environment variable not set")
     
-    # Login to Hugging Face before downloading the model (using torch.hub)
-    torch.hub._validate_not_a_forked_repo = lambda a, b, c: True
-    model, model_config = get_pretrained_model("stabilityai/stable-audio-open-1.0")
+    # Download and set up the model
+    model, model_config = get_pretrained_model("stabilityai/stable-audio-open-1.0", TOKEN=hf_token)
     sample_rate = model_config["sample_rate"]
     sample_size = model_config["sample_size"]
 
@@ -53,7 +52,7 @@ def generate_audio(prompt, seconds_total=30, steps=100, cfg_scale=7):
 
     # Peak normalize, clip, convert to int16
     output = output.to(torch.float32).div(torch.max(torch.abs(output))).clamp(-1, 1).mul(32767).to(torch.int16).cpu()
-    
+
     # Generate a unique filename for the output
     unique_filename = f"output_{uuid.uuid4().hex}.wav"
     # Save to file