Update app.py

app.py

@@ -13,6 +13,8 @@ pipeline = pipeline.to("cuda")
 
 def predict(mask_img):
     prompt = "a green frog, highly detailed, natural lighting"
+    # get size of the input image
+    size = mask_img["image"].size
     image = pipeline(prompt=prompt,
                      num_inference_steps=35,
                      image=mask_img["image"].convert("RGB"), 
@@ -20,7 +22,7 @@ def predict(mask_img):
                      guidance_scale=9
                     ).images[0]
      
-    return image
+    return image.resize(size)
 
 demo = gr.Interface(
     fn=predict,