Update demos/musicgen_colab.py

demos/musicgen_colab.py

@@ -200,7 +200,7 @@ class Predictor:
                         assert outputs_diffusion.shape[1] == 1  # output is mono
                         outputs_diffusion = rearrange(outputs_diffusion, '(s b) c t -> b (s c) t', s=2)
                     outputs_diffusion = outputs_diffusion.detach().cpu()
-                    return task_id, (output, outputs_diffusion) #Return the task id.
+                    return output, outputs_diffusion #Return the task id.
                 else:
                   return task_id, (output, None)
             except Exception as e: