Update demos/musicgen_app.py
Browse files- 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=
|
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)
|