fix more

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 from datasets import load_dataset
-from PIL import Image  
+from PIL import Image
 
 import re
 import os
@@ -11,23 +11,27 @@ from share_btn import community_icon_html, loading_icon_html, share_js
 word_list_dataset = load_dataset("Gustavosta/Stable-Diffusion-Prompts")
 word_list = word_list_dataset["train"]['Prompt']
 
-
 from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler
 import torch
 
 model_id = "stabilityai/stable-diffusion-2-1-base"
 
 scheduler = EulerDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
-pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, torch_dtype=torch.float16)
-
+pipe = StableDiffusionPipeline.from_pretrained(
+    model_id,
+    scheduler=scheduler,
+    safety_checker=None,
+    torch_dtype=torch.float16)
 
 is_gpu_busy = False
+
+
 def infer(prompt, negative, scale):
     global is_gpu_busy
     for filter in word_list:
         if re.search(rf"\b{filter}\b", prompt):
             raise gr.Error("Unsafe content found. Please try again with different prompts.")
-        
+
     images = []
     generator = torch.manual_seed(68781279)
     payload = {'prompt': prompt, 'negative_prompt': negative, 'guidance_scale': scale}
@@ -40,10 +44,10 @@ def infer(prompt, negative, scale):
                  width=768).images[0]
 
     images.append(image)
-    
+
     return images
-    
-    
+
+
 css = """
         .gradio-container {
             font-family: 'IBM Plex Sans', sans-serif;
@@ -198,7 +202,6 @@ examples = [
     ],
 ]
 
-
 with block:
     gr.HTML(
         """
@@ -298,19 +301,19 @@ with block:
         ).style(grid=[2], height="auto")
 
         with gr.Group(elem_id="container-advanced-btns"):
-            #advanced_button = gr.Button("Advanced options", elem_id="advanced-btn")
+            # advanced_button = gr.Button("Advanced options", elem_id="advanced-btn")
             with gr.Group(elem_id="share-btn-container"):
                 community_icon = gr.HTML(community_icon_html)
                 loading_icon = gr.HTML(loading_icon_html)
                 share_button = gr.Button("Share to community", elem_id="share-btn")
 
         with gr.Accordion("Advanced settings", open=False):
-        #    gr.Markdown("Advanced settings are temporarily unavailable")
-        #    samples = gr.Slider(label="Images", minimum=1, maximum=4, value=4, step=1)
-        #    steps = gr.Slider(label="Steps", minimum=1, maximum=50, value=45, step=1)
-             guidance_scale = gr.Slider(
+            #    gr.Markdown("Advanced settings are temporarily unavailable")
+            #    samples = gr.Slider(label="Images", minimum=1, maximum=4, value=4, step=1)
+            #    steps = gr.Slider(label="Steps", minimum=1, maximum=50, value=45, step=1)
+            guidance_scale = gr.Slider(
                 label="Guidance Scale", minimum=0, maximum=50, value=9, step=0.1
-             )
+            )
         #    seed = gr.Slider(
         #        label="Seed",
         #        minimum=0,
@@ -319,13 +322,14 @@ with block:
         #        randomize=True,
         #    )
 
-        ex = gr.Examples(examples=examples, fn=infer, inputs=[text, negative, guidance_scale], outputs=[gallery, community_icon, loading_icon, share_button], cache_examples=False)
+        ex = gr.Examples(examples=examples, fn=infer, inputs=[text, negative, guidance_scale],
+                         outputs=[gallery, community_icon, loading_icon, share_button], cache_examples=False)
         ex.dataset.headers = [""]
         negative.submit(infer, inputs=[text, negative, guidance_scale], outputs=[gallery], postprocess=False)
         text.submit(infer, inputs=[text, negative, guidance_scale], outputs=[gallery], postprocess=False)
         btn.click(infer, inputs=[text, negative, guidance_scale], outputs=[gallery], postprocess=False)
-        
-        #advanced_button.click(
+
+        # advanced_button.click(
         #    None,
         #    [],
         #    text,
@@ -334,13 +338,12 @@ with block:
         #        const options = document.querySelector("body > gradio-app").querySelector("#advanced-options");
         #        options.style.display = ["none", ""].includes(options.style.display) ? "flex" : "none";
         #    }""",
-        #)
+        # )
         share_button.click(
             None,
             [],
             [],
             _js=share_js,
         )
-        
 
-block.queue(concurrency_count=80, max_size=100).launch(max_threads=150)
+block.queue(concurrency_count=80, max_size=100).launch(max_threads=150)