Update app.py

app.py

@@ -6,7 +6,7 @@ import random
 import torch
 import spaces
 
-from diffusers import FlaxStableDiffusionInpaintPipeline
+from diffusers import StableDiffusionXLInpaintPipeline
 from PIL import Image, ImageFilter, ImageEnhance
 import PIL.ImageOps
 from diffusers.pipelines.stable_diffusion import safety_checker
@@ -24,10 +24,9 @@ else:
     device = "cpu"
     floatType = torch.float32
     variant = None
-pipe, params = FlaxStableDiffusionInpaintPipeline.from_pretrained("xvjiarui/stable-diffusion-2-inpainting", torch_dtype = floatType, variant = variant, safety_checker=None)
 
-#pipe = StableDiffusionXLInpaintPipeline.from_pretrained("diffusers/stable-diffusion-xl-1.0-inpainting-0.1", torch_dtype = floatType, variant = variant)
-#pipe = pipe.to(device)
+pipe = StableDiffusionXLInpaintPipeline.from_pretrained("diffusers/stable-diffusion-xl-1.0-inpainting-0.1", torch_dtype = floatType, variant = variant)
+pipe = pipe.to(device)
 
 def update_seed(is_randomize_seed, seed):
     if is_randomize_seed:
@@ -221,8 +220,9 @@ def inpaint_on_gpu(
         strength,
         denoising_steps
 ):
+
     return pipe(
-#        seeds = [seed],
+        seeds = [seed],
         width = process_width,
         height = process_height,
         prompt = prompt,