Update template/app_advanced.py

template/app_advanced.py

@@ -50,7 +50,7 @@ def txt_to_img(prompt, neg_prompt, guidance, steps, width, height, generator):
       height = height,
       generator = generator)
     
-    return result.images[0]#replace_nsfw_images(result)
+    return result.images[0]
 
 def img_to_img(prompt, neg_prompt, img, strength, guidance, steps, width, height, generator):
 
@@ -67,14 +67,7 @@ def img_to_img(prompt, neg_prompt, img, strength, guidance, steps, width, height
         height = height,
         generator = generator)
         
-    return result.images[0]#replace_nsfw_images(result)
-
-def replace_nsfw_images(results):
-
-    for i in range(len(results.images)):
-      if 'nsfw_content_detected' in results and results.nsfw_content_detected[i]:
-        results.images[i] = Image.open("nsfw.png")
-    return results.images[0]
+    return result.images[0]
 
 css = """.main-div div{display:inline-flex;align-items:center;gap:.8rem;font-size:1.75rem}.main-div div h1{font-weight:900;margin-bottom:7px}.main-div p{margin-bottom:10px;font-size:94%}a{text-decoration:underline}.tabs{margin-top:0;margin-bottom:0}#gallery{min-height:20rem}
 """