Update app.py

app.py

@@ -24,7 +24,7 @@ else:
     device = "cpu"
     floatType = torch.float32
     variant = None
-pipe, params = FlaxStableDiffusionInpaintPipeline.from_pretrained("xvjiarui/stable-diffusion-2-inpainting", torch_dtype = floatType, variant = variant)
+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)