Update app.py

app.py

@@ -48,7 +48,7 @@ def infer(source_img, prompt, guide, steps, seed, strength):
     
     for i, image in enumerate(images_list["images"]):
         if(images_list["nsfw_content_detected"][i]):
-            images.append(safe_image)
+            images.append(image)
         else:
             images.append(image)    
     return images