Update app.py

app.py

@@ -6,7 +6,7 @@ import random
 import torch
 import spaces
 
-from diffusers import StableDiffusionXLInpaintPipeline
+from diffusers import FlaxStableDiffusionInpaintPipeline
 from PIL import Image, ImageFilter, ImageEnhance
 import PIL.ImageOps
 from diffusers.pipelines.stable_diffusion import safety_checker
@@ -24,8 +24,9 @@ else:
     device = "cpu"
     floatType = torch.float32
     variant = None
+pipe, params = FlaxStableDiffusionInpaintPipeline.from_pretrained("xvjiarui/stable-diffusion-2-inpainting", torch_dtype = floatType, variant = varian)
 
-pipe = StableDiffusionXLInpaintPipeline.from_pretrained("diffusers/stable-diffusion-xl-1.0-inpainting-0.1", torch_dtype = floatType, variant = variant)
+#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):