Update app.py

app.py

@@ -62,6 +62,34 @@ def image_to_image(ip, image, prompt, neg_prompt, width, height, ip_scale, stren
     
     return images[0]
 
+@spaces.GPU(enable_queue=True)
+def inpaint(ip, image_editor, prompt, neg_prompt, width, height, ip_scale, strength, guidance, steps):
+    inpaint_pipeline.to("cuda")
+    
+    image = image_editor['background']
+    mask = image_editor['composite']
+    
+    ip.thumbnail((1024, 1024))
+    image.thumbnail((1024, 1024))
+    mask.thumbnail((1024, 1024))
+    
+    inpaint_pipeline.set_ip_adapter_scale(ip_scale)
+    
+    images = inpaint_pipeline(
+        prompt=prompt,
+        image=image,
+        mask_image=mask,
+        ip_adapter_image=ip,
+        negative_prompt=neg_prompt,
+        width=width,
+        height=height,
+        strength=strength,
+        guidance_scale=guidance,
+        num_inference_steps=steps,
+    ).images
+    
+    return images[0]
+
 
 with gr.Blocks() as demo:
     gr.Markdown("""
@@ -98,5 +126,6 @@ with gr.Blocks() as demo:
 
     text_button.click(text_to_image, inputs=[text_ip, text_prompt, neg_prompt, width_slider, height_slider, ip_scale_slider, strength_slider, guidance_slider, steps_slider], outputs=output_image)
     image_button.click(image_to_image, inputs=[image_ip, image_image, image_prompt, neg_prompt, width_slider, height_slider, ip_scale_slider, strength_slider, guidance_slider, steps_slider], outputs=output_image)
+    inpaint_button.click(inpaint, inputs=[inpaint_ip, inpaint_editor, inpaint_prompt, neg_prompt, width_slider, height_slider, ip_scale_slider, strength_slider, guidance_slider, steps_slider], outputs=output_image)
 
 demo.launch()