Update app.py

app.py

@@ -1,6 +1,6 @@
 import spaces
 import gradio as gr
-from diffusers import AutoPipelineForText2Image, AutoPipelineForImage2Image, AutoencoderKL
+from diffusers import AutoPipelineForText2Image, AutoPipelineForImage2Image, AutoPipelineForInpainting, AutoencoderKL
 from diffusers.utils import load_image
 import torch
 
@@ -14,6 +14,10 @@ image_pipeline = AutoPipelineForImage2Image.from_pretrained("stabilityai/stable-
 image_pipeline.load_ip_adapter("h94/IP-Adapter", subfolder="sdxl_models", weight_name="ip-adapter_sdxl.bin")
 image_pipeline.set_ip_adapter_scale(0.6)
 
+inpaint_pipeline = AutoPipelineForInpainting.from_pretrained("diffusers/stable-diffusion-xl-1.0-inpainting-0.1", vae=vae, torch_dtype=torch.float16, variant="fp16", use_safetensors=True).to("cuda")
+inpaint_pipeline.load_ip_adapter("h94/IP-Adapter", subfolder="sdxl_models", weight_name="ip-adapter_sdxl.bin")
+inpaint_pipeline.set_ip_adapter_scale(0.6)
+
 @spaces.GPU(enable_queue=True)
 def text_to_image(ip, prompt, neg_prompt, width, height, ip_scale, strength, guidance, steps):
     text_pipeline.to("cuda")