Update edit_space.py

edit_space.py

@@ -331,8 +331,8 @@ class KontextEditModel():
         self.clear_cache()
 
         result_tensor = self._pil_to_tensor(result_pil)
-        final_image = self.blender.blend_inpaint(result_tensor, original_image_tensor, original_mask, kernel=10, sigma=10)[0]
-        # final_image = result_tensor
+        # final_image = self.blender.blend_inpaint(result_tensor, original_image_tensor, original_mask, kernel=10, sigma=10)[0]
+        final_image = result_tensor
         return (final_image, self._pil_to_tensor(spatial_pil), original_mask)
 
     def local_edit(self,
@@ -366,8 +366,8 @@ class KontextEditModel():
 
         self.clear_cache()
         result_tensor = self._pil_to_tensor(result_pil)
-        final_image = self.blender.blend_inpaint(result_tensor, original_image_tensor, original_mask, kernel=10, sigma=10)[0]
-        # final_image = result_tensor
+        # final_image = self.blender.blend_inpaint(result_tensor, original_image_tensor, original_mask, kernel=10, sigma=10)[0]
+        final_image = result_tensor
         return (final_image, self._pil_to_tensor(spatial_pil), original_mask)
 
     def foreground_edit(self,