Update app.py

app.py

@@ -11,8 +11,7 @@ scheduler = DPMSolverMultistepScheduler.from_pretrained(model_id, subfolder="sch
 pipe = StableDiffusionPipeline.from_pretrained(
   model_id,
   torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
-  scheduler=scheduler,
-  pipe.safety_checker=lambda )
+  scheduler=scheduler)
 
 pipe_i2i = StableDiffusionImg2ImgPipeline.from_pretrained(
   model_id,
@@ -96,6 +95,11 @@ def img_to_img(prompt, neg_prompt, img, strength, guidance, steps, width, height
         
     return result.images[0]
 
+    def fake_safety_checker(images, **kwargs):
+      return result.images[0], [False] * len(images)
+    
+    pipe.safety_checker = fake_safety_checker
+
 css = """.main-div div{display:inline-flex;align-items:center;gap:.8rem;font-size:1.75rem}.main-div div h1{font-weight:900;margin-bottom:7px}.main-div p{margin-bottom:10px;font-size:94%}a{text-decoration:underline}.tabs{margin-top:0;margin-bottom:0}#gallery{min-height:20rem}
 """
 with gr.Blocks(css=css) as demo: