ford442 commited on
Commit
82e7e60
·
verified ·
1 Parent(s): 10b7203

Update demos/musicgen_colab.py

Browse files
Files changed (1) hide show
  1. 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."""