Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 from diffusers import StableDiffusionPipeline, StableDiffusionImg2ImgPipeline
+from diffusers import StableDiffusionInpaintPipeline
 import torch
 from PIL import Image, ImageDraw
 import time
@@ -70,14 +71,13 @@ def load_txt2img():
 def load_img2img():
     global pipe_img2img
     if pipe_img2img is None:
-        print("Loading Image-to-Image model...")
-        pipe_img2img = StableDiffusionImg2ImgPipeline.from_pretrained(
+        print("Loading Inpainting model...")
+        pipe_img2img = StableDiffusionInpaintPipeline.from_pretrained(
             "runwayml/stable-diffusion-inpainting",
             torch_dtype=torch_dtype,
-            use_safetensors=True,
-            safety_checker=None,
-            #requires_safety_checker=False
+            safety_checker=None
         ).to(device)
+
         
         from diffusers import DPMSolverMultistepScheduler
         pipe_img2img.scheduler = DPMSolverMultistepScheduler.from_config(