Update demos/musicgen_colab.py
Browse files- demos/musicgen_colab.py +1 -1
demos/musicgen_colab.py
CHANGED
|
@@ -210,7 +210,7 @@ class Predictor:
|
|
| 210 |
self.current_task_id += 1
|
| 211 |
task = (self.current_task_id, text, melody, duration, use_diffusion, gen_params)
|
| 212 |
self.task_queue.put(task)
|
| 213 |
-
return self.current_task_id
|
| 214 |
|
| 215 |
def get_result(self, task_id):
|
| 216 |
"""Retrieves the result of a prediction task."""
|
|
|
|
| 210 |
self.current_task_id += 1
|
| 211 |
task = (self.current_task_id, text, melody, duration, use_diffusion, gen_params)
|
| 212 |
self.task_queue.put(task)
|
| 213 |
+
return self.current_task_id, (None, None)
|
| 214 |
|
| 215 |
def get_result(self, task_id):
|
| 216 |
"""Retrieves the result of a prediction task."""
|