Update app.py

app.py

@@ -45,6 +45,8 @@ def process(input_image_editor,
     image = input_image_editor['background']
     mask = input_image_editor['layers'][0]
     size = (768, 768)
+    image_or = image.copy()
+    
     image = image.convert("RGB").resize(size)
     mask = mask.convert("RGB").resize(size)
     generator = torch.Generator(device="cuda").manual_seed(seed)
@@ -62,7 +64,7 @@ def process(input_image_editor,
     true_guidance_scale=true_guidance_scale
     ).images[0]
 
-    return image
+    return result.resize((image_or.size[:2]))
 
 with gr.Blocks() as demo:
     gr.Markdown(MARKDOWN)