Update main.py

main.py

@@ -82,6 +82,7 @@ async def image_to_image(
         strength=strength,
         guidance_scale=0.0,
     ).images[0]
+    image = image.resize(init_image.size)
 
     bytes = BytesIO()
     image.save(bytes, "PNG")
@@ -113,8 +114,9 @@ async def inpainting(
         strength=strength,
         guidance_scale=0.0,
     ).images[0]
+    image = image.resize(init_image.size)
 
     bytes = BytesIO()
     image.save(bytes, "PNG")
     bytes.seek(0)
-    return StreamingResponse(bytes, media_type="image/png")
+    return StreamingResponse(bytes, media_type="image/png")