abidlabs HF staff commited on
Commit
8d7eac5
1 Parent(s): dd22113

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -177,7 +177,7 @@ class Model:
177
  127.5).cpu().numpy().clip(0, 255).astype(np.uint8)
178
 
179
  results = [x_samples[i] for i in range(num_samples)]
180
- return [255 - detected_map] + results
181
 
182
  @torch.inference_mode()
183
  def process_hough(self, input_image, prompt, a_prompt, n_prompt,
 
177
  127.5).cpu().numpy().clip(0, 255).astype(np.uint8)
178
 
179
  results = [x_samples[i] for i in range(num_samples)]
180
+ return [input_image] + results
181
 
182
  @torch.inference_mode()
183
  def process_hough(self, input_image, prompt, a_prompt, n_prompt,