sonalkum commited on
Commit
6798398
·
verified ·
1 Parent(s): 31ef813

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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(duration=120) # 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}")
 
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}")