Update app.py

app.py

@@ -67,6 +67,7 @@ def txt_to_img(prompt, neg_prompt, guidance, steps, width, height, generator):
 
     result = pipe(
       prompt,
+      safety_checker = none,
       negative_prompt = neg_prompt,
       num_inference_steps = int(steps),
       guidance_scale = guidance,
@@ -74,7 +75,7 @@ def txt_to_img(prompt, neg_prompt, guidance, steps, width, height, generator):
       height = height,
       generator = generator)
     
-    return result.imagesimage = pipe(prompt, safety_checker=None).images[0]
+    return result.images[0]
 
 def img_to_img(prompt, neg_prompt, img, strength, guidance, steps, width, height, generator):
 
@@ -82,6 +83,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,
         negative_prompt = neg_prompt,
         init_image = img,
         num_inference_steps = int(steps),