Update edit_space.py

edit_space.py

@@ -269,7 +269,8 @@ class KontextEditModel():
             }
 
         # Prepare debug/output images
-        debug_image = lineart_output if lineart_output is not None else self.color_processor(colored_image, detect_resolution=1024)
+        colored_image_np = np.array(self._tensor_to_pil(colored_image))
+        debug_image = lineart_output if lineart_output is not None else self.color_processor(colored_image_np, detect_resolution=1024, output_type="pil")
 
         # Run inference
         result_pil = self.pipe(