Update main.py

main.py

@@ -67,11 +67,11 @@ async def text_to_image(
 async def image_to_image(
     request: Request,
     prompt: str = Form(...),
-    init_bytes: bytes = Form(...),
+    init_image: UploadFile = File(...),
     num_inference_steps: int = Form(2),
     strength: float = Form(1.0),
 ):
-    # init_bytes = await init_image.read()
+    init_bytes = await init_image.read()
     init_image = Image.open(BytesIO(init_bytes))
     init_image = init_image.convert("RGB").resize((512, 512))
 
@@ -93,15 +93,15 @@ async def image_to_image(
 async def inpainting(
     request: Request,
     prompt: str = Form(...),
-    init_bytes: bytes = Form(...),
-    mask_bytes: bytes = Form(...),
+    init_image: UploadFile = File(...),
+    mask_image: UploadFile = File(...),
     num_inference_steps: int = Form(2),
     strength: float = Form(1.0),
 ):
-    # init_bytes = await init_image.read()
+    init_bytes = await init_image.read()
     init_image = Image.open(BytesIO(init_bytes))
     init_image = init_image.convert("RGB").resize((512, 512))
-    # mask_bytes = await mask_image.read()
+    mask_bytes = await mask_image.read()
     mask_image = Image.open(BytesIO(mask_bytes))
     mask_image = mask_image.convert("RGB").resize((512, 512))