ysharma HF staff commited on
Commit
1f6cc31
·
1 Parent(s): 21f2c0a
Files changed (1) hide show
  1. model.py +2 -1
model.py CHANGED
@@ -631,7 +631,8 @@ class Model:
631
  127.5).cpu().numpy().clip(0, 255).astype(np.uint8)
632
 
633
  results = [x_samples[i] for i in range(num_samples)]
634
- return [detected_map] + results, [detected_map] + results
 
635
 
636
  @torch.inference_mode()
637
  def process_depth(self, input_image, prompt, a_prompt, n_prompt,
 
631
  127.5).cpu().numpy().clip(0, 255).astype(np.uint8)
632
 
633
  results = [x_samples[i] for i in range(num_samples)]
634
+
635
+ return [detected_map] + results
636
 
637
  @torch.inference_mode()
638
  def process_depth(self, input_image, prompt, a_prompt, n_prompt,