Upload inference.py

inference.py

@@ -84,7 +84,7 @@ def inference(pipe, img1, img2, num_steps):
        num_images_per_prompt=1
     ).images
     
-    edited_image = edited_image[0]
+    edited_image = edited_image[0].convert("L")
     
     return edited_image