Update app.py

app.py

@@ -45,7 +45,8 @@ def set_base_models():
     img2img = StableDiffusionImg2ImgPipeline.from_pretrained(
         "Lykon/dreamshaper-8",
         torch_dtype = torch.float16,
-        variant = "fp16"
+        variant = "fp16",
+        safety_checker=None
     )
     img2img.to("cuda")
     return txt2img, img2img