Update app.py

app.py

@@ -11,7 +11,7 @@ import torch
 from torch import autocast
 import cv2
 from matplotlib import pyplot as plt
-from diffusers import StableDiffusionInpaintPipeline
+from diffusers import DiffusionPipeline
 from torchvision import transforms
 from clipseg.models.clipseg import CLIPDensePredT
 
@@ -22,7 +22,7 @@ def download_image(url):
     return PIL.Image.open(BytesIO(response.content)).convert("RGB")
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
-pipe = StableDiffusionInpaintingPipeline.from_pretrained(
+pipe = DiffusionPipeline.from_pretrained(
     "runwayml/stable-diffusion-inpainting",
     revision="fp16", 
     torch_dtype=torch.float16,