Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -21,7 +21,7 @@ def load_model():
|
|
21 |
model, model_config = load_model()
|
22 |
|
23 |
# Function to set up, generate, and process the audio
|
24 |
-
@spaces.GPU
|
25 |
def generate_audio(prompt, seconds_total=10, steps=100, cfg_scale=7):
|
26 |
print(f"Prompt received: {prompt}")
|
27 |
print(f"Settings: Duration={seconds_total}s, Steps={steps}, CFG Scale={cfg_scale}")
|
|
|
21 |
model, model_config = load_model()
|
22 |
|
23 |
# Function to set up, generate, and process the audio
|
24 |
+
@spaces.GPU # Allocate GPU only when this function is called
|
25 |
def generate_audio(prompt, seconds_total=10, steps=100, cfg_scale=7):
|
26 |
print(f"Prompt received: {prompt}")
|
27 |
print(f"Settings: Duration={seconds_total}s, Steps={steps}, CFG Scale={cfg_scale}")
|