Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -7,6 +7,10 @@ from stable_audio_tools.inference.generation import generate_diffusion_cond
|
|
7 |
|
8 |
def gen_music(description):
|
9 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
|
|
|
|
|
|
|
10 |
|
11 |
# Download model
|
12 |
model, model_config = get_pretrained_model("stabilityai/stable-audio-open-1.0")
|
|
|
7 |
|
8 |
def gen_music(description):
|
9 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
10 |
+
|
11 |
+
# Fetch the Hugging Face token from the environment variable
|
12 |
+
hf_token = os.getenv('HF_TOKEN')
|
13 |
+
print(f"Hugging Face token: {hf_token}")
|
14 |
|
15 |
# Download model
|
16 |
model, model_config = get_pretrained_model("stabilityai/stable-audio-open-1.0")
|