Update main.py

main.py

@@ -49,9 +49,7 @@ async def root():
 
 @app.post("/text-to-image/")
 async def text_to_image(
-    request: Request,
-    prompt: str,
-    num_inference_steps: int = 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
@@ -66,13 +64,13 @@ async def text_to_image(
 @app.post("/image-to-image/")
 async def image_to_image(
     request: Request,
-    prompt: str,
-    init_bytes: bytes,
-    num_inference_steps: int = 2,
-    strength: float = 1.0,
+    prompt: str = Form(...),
+    init_image: UploadFile = File(...),
+    num_inference_steps: int = Form(2),
+    strength: float = Form(0.5),
 ):
-    # init_bytes = await init_image.read()
-    init_image = Image.open(BytesIO(init_bytes))
+    bytes = await init_image.read()
+    init_image = Image.open(BytesIO(bytes))
     init_image = init_image.convert("RGB").resize((512, 512))
 
     image = request.state.img2img(
@@ -92,17 +90,17 @@ async def image_to_image(
 @app.post("/inpainting/")
 async def inpainting(
     request: Request,
-    prompt: str,
-    init_bytes: bytes,
-    mask_bytes: bytes,
-    num_inference_steps: int = 2,
-    strength: float = 1.0,
+    prompt: str = Form(...),
+    init_image: UploadFile = File(...),
+    mask_image: UploadFile = File(...),
+    num_inference_steps: int = Form(3),
+    strength: float = Form(0.5),
 ):
-    # init_bytes = await init_image.read()
-    init_image = Image.open(BytesIO(init_bytes))
+    bytes = await init_image.read()
+    init_image = Image.open(BytesIO(bytes))
     init_image = init_image.convert("RGB").resize((512, 512))
-    # mask_bytes = await mask_image.read()
-    mask_image = Image.open(BytesIO(mask_bytes))
+    bytes = await mask_image.read()
+    mask_image = Image.open(BytesIO(bytes))
     mask_image = mask_image.convert("RGB").resize((512, 512))
 
     image = request.state.inpaint(