ameerazam08 commited on
Commit
bb23a85
1 Parent(s): 4ca91f3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ def generate_audio(prompt, bpm, seconds_total):
12
  device = "cuda" if torch.cuda.is_available() else "cpu"
13
 
14
  # Download model
15
- model, model_config = get_pretrained_model("stabilityai/stable-audio-open-1.0")
16
  sample_rate = model_config["sample_rate"]
17
  sample_size = model_config["sample_size"]
18
 
 
12
  device = "cuda" if torch.cuda.is_available() else "cpu"
13
 
14
  # Download model
15
+ model, model_config = get_pretrained_model("stabilityai/stable-audio-open-1.0",token=os.environ.get('HF_TOKEN'))
16
  sample_rate = model_config["sample_rate"]
17
  sample_size = model_config["sample_size"]
18