Update app.py

app.py

@@ -11,7 +11,7 @@ import random
 start_time = time.time()
 current_steps = 15
 
-pipe = DiffusionPipeline.from_pretrained("timbrooks/instruct-pix2pix", torch_dtype=torch.float16, safety_checker=None)
+pipe = DiffusionPipeline.from_pretrained("timbrooks/instruct-pix2pix", torch_dtype=torch.float16) # safety_checker=None)
 pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
 pipe.enable_xformers_memory_efficient_attention()
 pipe.unet.to(memory_format=torch.channels_last)
@@ -64,7 +64,7 @@ def inference(
         )
 
         # return replace_nsfw_images(result)
-        return result.images, f"Done. Seed: {seed}"
+        return result.images, results.nsfw_content_detected, f"Done. Seed: {seed}"
     except Exception as e:
         return None, error_str(e)