ford442 commited on
Commit
6e65b60
·
verified ·
1 Parent(s): f80309c

Update demos/musicgen_app.py

Browse files
Files changed (1) hide show
  1. demos/musicgen_app.py +1 -1
demos/musicgen_app.py CHANGED
@@ -253,7 +253,7 @@ class Predictor:
253
 
254
  _default_model_name = "facebook/musicgen-melody"
255
 
256
- @spaces.GPU(duration=60) # Use the decorator for Spaces
257
  def predict_full(model, model_path, use_mbd, text, melody, duration, topk, topp, temperature, cfg_coef):
258
  # Initialize Predictor *INSIDE* the function
259
  predictor = Predictor(model)
 
253
 
254
  _default_model_name = "facebook/musicgen-melody"
255
 
256
+ @spaces.GPU(duration=90) # Use the decorator for Spaces
257
  def predict_full(model, model_path, use_mbd, text, melody, duration, topk, topp, temperature, cfg_coef):
258
  # Initialize Predictor *INSIDE* the function
259
  predictor = Predictor(model)