Update app.py

app.py

@@ -63,11 +63,17 @@ def inference(prompt, guidance, steps, width=512, height=512, seed=0, img=None,
   except Exception as e:
     return None, error_str(e)
 
+    def dummy(images, **kwargs):
+        return images, False
+    
+
+It is similar to the notebook code, but this one works for local run
+
 def txt_to_img(prompt, neg_prompt, guidance, steps, width, height, generator):
 
     result = pipe(
       prompt,
-      safety_checker = none,
+      check_safety = dummy,
       negative_prompt = neg_prompt,
       num_inference_steps = int(steps),
       guidance_scale = guidance,
@@ -83,7 +89,7 @@ def img_to_img(prompt, neg_prompt, img, strength, guidance, steps, width, height
     img = img.resize((int(img.width * ratio), int(img.height * ratio)), Image.LANCZOS)
     result = pipe_i2i(
         prompt,
-        safety_checker = none,
+        check_safety = dummy,
         negative_prompt = neg_prompt,
         init_image = img,
         num_inference_steps = int(steps),