amazonaws-la commited on
Commit
b70be97
·
verified ·
1 Parent(s): 02b9bfc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -14,8 +14,7 @@ from PIL import Image
14
  from io import BytesIO
15
  from diffusers import StableDiffusionImg2ImgPipeline, AutoencoderKL, DiffusionPipeline
16
  from diffusers.utils import load_image
17
- from diffusers import StableDiffusionPipeline
18
- from diffusers import logging
19
 
20
  DESCRIPTION = "# SDXL"
21
  if not torch.cuda.is_available():
 
14
  from io import BytesIO
15
  from diffusers import StableDiffusionImg2ImgPipeline, AutoencoderKL, DiffusionPipeline
16
  from diffusers.utils import load_image
17
+ from diffusers.pipelines.stable_diffusion import StableDiffusionSafetyChecker
 
18
 
19
  DESCRIPTION = "# SDXL"
20
  if not torch.cuda.is_available():