add inpaint

app.py

@@ -50,7 +50,7 @@ def prepare_image_and_mask(
     return background, mask
 
 
-def inpaint(
+def outpaint(
     image,
     padding_top=0,
     padding_bottom=0,
@@ -79,6 +79,31 @@ def inpaint(
     return result
 
 
+def inpaint(
+    image,
+    mask,
+    prompt="",
+    num_inference_steps=28,
+    guidance_scale=50,
+):
+    background = image.convert("RGB")
+    mask = mask.convert("L")
+
+    result = pipe(
+        prompt=prompt,
+        height=background.height,
+        width=background.width,
+        image=background,
+        mask_image=mask,
+        num_inference_steps=num_inference_steps,
+        guidance_scale=guidance_scale,
+    ).images[0]
+
+    result = result.convert("RGBA")
+
+    return result
+
+
 def rmbg(image=None, url=None):
     if image is None:
         image = url
@@ -102,6 +127,8 @@ def main(*args, progress=gr.Progress(track_tqdm=True)):
     if api_num == 1:
         return rmbg(*args)
     elif api_num == 2:
+        return outpaint(*args)
+    elif api_num == 3:
         return inpaint(*args)
 
 
@@ -131,9 +158,25 @@ outpaint_tab = gr.Interface(
     api_name="outpainting",
 )
 
+
+inpaint_tab = gr.Interface(
+    fn=main,
+    inputs=[
+        gr.Number(3, visible=False),
+        gr.Image(label="image"),
+        gr.Image(label="mask"),
+        gr.Text(label="prompt"),
+        gr.Number(value=50, label="num_inference_steps"),
+        gr.Number(value=28, label="guidance_scale"),
+    ],
+    outputs=["image"],
+    api_name="inpaint",
+    examples=[[3, "./assets/rocket.png", "./assets/Inpainting mask.png"]],
+)
+
 demo = gr.TabbedInterface(
-    [rmbg_tab, outpaint_tab],
-    ["remove background", "outpainting"],
+    [rmbg_tab, outpaint_tab, inpaint_tab],
+    ["remove background", "outpainting", "inpainting"],
     title="Utilities that require GPU",
 )