Upload inference.py

inference.py

@@ -103,4 +103,6 @@ def inference(model_id,device, img1, img2):
        num_images_per_prompt=1
     ).images
     
+    edited_image = edited_image[0]
+    
     return edited_image