Update main.py

main.py

@@ -49,7 +49,9 @@ async def root():
 
 @app.post("/text-to-image/")
 async def text_to_image(
-    request: Request, prompt: str = Form(...), num_inference_steps: int = Form(1)
+    request: Request,
+    prompt: str = Form(...),
+    num_inference_steps: int = Form(1),
 ):
     image = request.state.text2img(
         prompt=prompt, num_inference_steps=num_inference_steps, guidance_scale=0.0
@@ -65,12 +67,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),
+    strength: float = Form(1.0),
 ):
-    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 +93,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),
-    strength: float = Form(0.5),
+    init_bytes: bytes = Form(...),
+    mask_bytes: bytes = Form(...),
+    num_inference_steps: int = Form(2),
+    strength: float = Form(1.0),
 ):
-    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(