Update main.py

main.py

@@ -65,12 +65,12 @@ async def text_to_image(
 async def image_to_image(
     request: Request,
     prompt: str = Form(...),
-    init_image: UploadFile = File(...),
+    init_bytes: bytes = Form(...),
     num_inference_steps: int = Form(2),
     strength: float = Form(0.5),
 ):
-    bytes = await init_image.read()
-    init_image = Image.open(BytesIO(bytes))
+    # init_bytes = await init_image.read()
+    init_image = Image.open(BytesIO(init_bytes))
     init_image = init_image.convert("RGB").resize((512, 512))
 
     image = request.state.img2img(
@@ -91,16 +91,16 @@ async def image_to_image(
 async def inpainting(
     request: Request,
     prompt: str = Form(...),
-    init_image: UploadFile = File(...),
-    mask_image: UploadFile = File(...),
-    num_inference_steps: int = Form(3),
+    init_bytes: bytes = Form(...),
+    mask_bytes: bytes = Form(...),
+    num_inference_steps: int = Form(2),
     strength: float = Form(0.5),
 ):
-    bytes = await init_image.read()
-    init_image = Image.open(BytesIO(bytes))
+    # init_bytes = await init_image.read()
+    init_image = Image.open(BytesIO(init_bytes))
     init_image = init_image.convert("RGB").resize((512, 512))
-    bytes = await mask_image.read()
-    mask_image = Image.open(BytesIO(bytes))
+    # mask_bytes = await mask_image.read()
+    mask_image = Image.open(BytesIO(mask_bytes))
     mask_image = mask_image.convert("RGB").resize((512, 512))
 
     image = request.state.inpaint(