Update app.py

app.py

@@ -105,7 +105,7 @@ def process(
         num_inference_steps=num_inference_steps_slider
         )
     print('INFERENCE DONE')
-    return result.images[0], result.images[1]
+    return result.images[0].resize((image.size), Image.LANCZOS), result.images[1].resize((image.size), Image.LANCZOS)
 
      #resized_mask = mask.resize((width, height), Image.LANCZOS)