Update app.py

app.py

@@ -29,12 +29,8 @@ def infer(prompt):
     #image = pipe(prompt, init_image=init_image)["sample"][0]
     images_list = pipe([prompt] * 1, guidance_scale=7.5, num_inference_steps=25, height=512, width=512)    #TODO
     images = []
-    safe_image = Image.open(r"unsafe.png")
     for i, image in enumerate(images_list["images"]):
-        if(images_list["nsfw_content_detected"][i]):
-            images.append(safe_image)
-        else:
-            images.append(image)
+        images.append(image)
     
     return images