Update app.py

app.py

@@ -1,14 +1,7 @@
 from diffusers import AutoPipelineForText2Image
-import torch
 import random
-import os
 import gradio as gr
 
-hf_token = os.getenv("HF_TOKEN")
-nsfw_filter = int(os.getenv("Safe"))
-
-naughtyWords = ["nude", "nsfw", "naked", "porn", "boob", "tit", "nipple", "vagina", "pussy", "panties", "underwear", "upskirt", "bottomless", "topless", "petite", "xxx"]
-
 css = """
 .btn-green {
   background-image: linear-gradient(to bottom right, #6dd178, #00a613) !important;
@@ -21,15 +14,6 @@ css = """
 """
 
 def generate(prompt, samp_steps, batch_size, seed, progress=gr.Progress(track_tqdm=True)):
-    prompt = prompt.lower()
-    if nsfw_filter:
-        if prompt[:10] == "krebzonide":
-            prompt = prompt[10:]
-        else:
-            neg_prompt = neg_prompt + ", child, nsfw, nipples, nude, underwear, naked"
-            for word in naughtyWords:
-                if prompt.find(word) >= 0:
-                    return None, 80085
     if seed < 0:
         seed = random.randint(1,999999)
     images = pipe(
@@ -45,8 +29,7 @@ def set_base_model():
     pipe = AutoPipelineForText2Image.from_pretrained(
         "stabilityai/sdxl-turbo",
         torch_dtype = torch.float16,
-        variant = "fp16",
-        #use_auth_token=hf_token
+        variant = "fp16"
     )
     pipe.to("cuda")
     return pipe
@@ -55,13 +38,16 @@ with gr.Blocks(css=css) as demo:
     with gr.Column():
         prompt = gr.Textbox(label="Prompt")
         submit_btn = gr.Button("Generate", elem_classes="btn-green")
+        
         with gr.Row():
-            samp_steps = gr.Slider(1, 5, value=1, step=1, label="Sampling steps")
-            batch_size = gr.Slider(1, 6, value=1, step=1, label="Batch size", interactive=True)
+            sampling_steps = gr.Slider(1, 4, value=1, step=1, label="Sampling steps")
+            batch_size = gr.Slider(1, 6, value=1, step=1, label="Batch size")
             seed = gr.Number(label="Seed", value=-1, minimum=-1, precision=0)
             lastSeed = gr.Number(label="Last Seed", value=-1, interactive=False)
-        gallery = gr.Gallery(show_label=False, preview=True, container=False, height=700)
-    submit_btn.click(generate, [prompt, samp_steps, batch_size, seed], [gallery, lastSeed], queue=True)
+            
+        gallery = gr.Gallery(show_label=False, preview=True, container=False, height=650)
+        
+    submit_btn.click(generate, [prompt, sampling_steps, batch_size, seed], [gallery, lastSeed], queue=True)
     
 pipe = set_base_model()
 demo.launch(debug=True)