Update app.py

app.py

@@ -183,8 +183,6 @@ def txt_to_img(model_path, prompt, n_images, neg_prompt, guidance, steps, width,
       callback=pipe_callback)
 
     # update_state(f"Done. Seed: {seed}")
-    
-    return replace_nsfw_images(result)
 
 def img_to_img(model_path, prompt, n_images, neg_prompt, img, strength, guidance, steps, width, height, generator, seed):
 
@@ -237,17 +235,6 @@ def img_to_img(model_path, prompt, n_images, neg_prompt, img, strength, guidance
 
     # update_state(f"Done. Seed: {seed}")
         
-    return replace_nsfw_images(result)
-
-def replace_nsfw_images(results):
-
-    if is_colab:
-      return results.images
-      
-    for i in range(len(results.images)):
-      if results.nsfw_content_detected[i]:
-        results.images[i] = Image.open("nsfw.png")
-    return results.images
 
 # css = """.finetuned-diffusion-div div{display:inline-flex;align-items:center;gap:.8rem;font-size:1.75rem}.finetuned-diffusion-div div h1{font-weight:900;margin-bottom:7px}.finetuned-diffusion-div p{margin-bottom:10px;font-size:94%}a{text-decoration:underline}.tabs{margin-top:0;margin-bottom:0}#gallery{min-height:20rem}
 # """
@@ -332,16 +319,7 @@ with gr.Blocks(css="style.css") as demo:
     ], inputs=[model_name, prompt, guidance, steps], outputs=outputs, fn=inference, cache_examples=False)
 
     gr.HTML("""
-    <div style="border-top: 1px solid #303030;">
-      <br>
-      <p>Models by <a href="https://huggingface.co/nitrosocke">@nitrosocke</a>, <a href="https://twitter.com/haruu1367">@haruu1367</a>, <a href="https://twitter.com/DGSpitzer">@Helixngc7293</a>, <a href="https://twitter.com/dal_mack">@dal_mack</a>, <a href="https://twitter.com/prompthero">@prompthero</a> and others. ❤️</p>
-      <p>This space uses the <a href="https://github.com/LuChengTHU/dpm-solver">DPM-Solver++</a> sampler by <a href="https://arxiv.org/abs/2206.00927">Cheng Lu, et al.</a>.</p>
-      <p>Space by:<br>
-      <a href="https://twitter.com/hahahahohohe"><img src="https://img.shields.io/twitter/follow/hahahahohohe?label=%40anzorq&style=social" alt="Twitter Follow"></a><br>
-      <a href="https://github.com/qunash"><img alt="GitHub followers" src="https://img.shields.io/github/followers/qunash?style=social" alt="Github Follow"></a></p><br><br>
-      <a href="https://www.buymeacoffee.com/anzorq" target="_blank"><img src="https://cdn.buymeacoffee.com/buttons/v2/default-yellow.png" alt="Buy Me A Coffee" style="height: 45px !important;width: 162px !important;" ></a><br><br>
-      <p><img src="https://visitor-badge.glitch.me/badge?page_id=anzorq.finetuned_diffusion" alt="visitors"></p>
-    </div>
+  
     """)
 
     demo.load(update_state_info, inputs=state_info, outputs=state_info, every=0.5, show_progress=False)