Update demos/musicgen_app.py
Browse files- demos/musicgen_app.py +1 -0
demos/musicgen_app.py
CHANGED
|
@@ -171,6 +171,7 @@ def predict_full(secret_token, model, model_path, decoder, text, melody, duratio
|
|
| 171 |
raise gr.Error(
|
| 172 |
f'Invalid secret token. Please fork the original space if you want to use it for yourself.')
|
| 173 |
|
|
|
|
| 174 |
global INTERRUPTING
|
| 175 |
global USE_DIFFUSION
|
| 176 |
INTERRUPTING = False
|
|
|
|
| 171 |
raise gr.Error(
|
| 172 |
f'Invalid secret token. Please fork the original space if you want to use it for yourself.')
|
| 173 |
|
| 174 |
+
print(f"generating {duration} sec of music for prompt: {text}")
|
| 175 |
global INTERRUPTING
|
| 176 |
global USE_DIFFUSION
|
| 177 |
INTERRUPTING = False
|