Update app.py

app.py

@@ -11,7 +11,8 @@ 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)
+  scheduler=scheduler,
+  pipe.safety_checker=lambda images, clip_input: (images, False))
 
 pipe_i2i = StableDiffusionImg2ImgPipeline.from_pretrained(
   model_id,
@@ -63,7 +64,7 @@ def inference(prompt, guidance, steps, width=512, height=512, seed=0, img=None,
   except Exception as e:
     return None, error_str(e)
       
-      pipe.safety_checker = lambda images, clip_input: (images, False)
+      
 
 def txt_to_img(prompt, neg_prompt, guidance, steps, width, height, generator):