Update app.py

app.py

@@ -3,7 +3,7 @@ import torch
 import numpy as np
 from PIL import Image
 from diffusers.models import AutoencoderKL
-from diffusers import AutoPipelineForInpainting, UNet2DConditionModel, DiffusionPipeline
+from diffusers import AutoPipelineForInpainting, UNet2DConditionModel, DiffusionPipeline, StableDiffusionInpaintPipeline
 import diffusers
 from share_btn import community_icon_html, loading_icon_html, share_js
 
@@ -21,7 +21,7 @@ def read_content(file_path: str) -> str:
 
 def predict(dict, prompt="", negative_prompt="", guidance_scale=7.5, steps=30, strength=0.8,model="Realistic_V5.0", scheduler="DPMSolverMultistepScheduler-Karras"):
     
-    pipe = AutoPipelineForInpainting.from_pretrained("SG161222/Realistic_Vision_V5.0_noVAE",vae=vae).to("cuda")
+    pipe = StableDiffusionInpaintPipeline.from_pretrained("SG161222/Realistic_Vision_V5.0_noVAE",vae=vae).to("cuda")
     
     if model == "Realistic_V5.1":
         pipe = AutoPipelineForInpainting.from_pretrained("SG161222/Realistic_Vision_V5.1_noVAE", vae=vae).to("cuda")