Update app.py

app.py

@@ -2,10 +2,7 @@ from diffusers import StableDiffusionXLPipeline, AutoencoderKL
 import torch
 import random
 import os
-#from controlnet_aux import OpenposeDetector
-#from diffusers.utils import load_image
 import gradio as gr
-import gc
 
 model_id = int(os.getenv("Model"))
 
@@ -44,15 +41,14 @@ css = """
 }
 """
 
-def generate(prompt, neg_prompt, samp_steps, guide_scale, batch_size, seed, height, width, progress=gr.Progress(track_tqdm=True)):
+def generate(prompt, neg_prompt, samp_steps, cfg_scale, batch_size, seed, height, width, progress=gr.Progress(track_tqdm=True)):
     if seed < 0:
         seed = random.randint(1,999999)
     images = pipe(
         prompt,
         negative_prompt=neg_prompt,
         num_inference_steps=samp_steps,
-        guidance_scale=guide_scale,
-        #cross_attention_kwargs={"scale": lora_scale},
+        guidance_scale=cfg_scale,
         num_images_per_prompt=batch_size,
         height=height,
         width=width,
@@ -105,17 +101,17 @@ with gr.Blocks(css=css) as demo:
         submit_btn = gr.Button("Generate", elem_classes="btn-green")
         with gr.Row():
             samp_steps = gr.Slider(1, 50, value=20, step=1, label="Sampling steps")
-            guide_scale = gr.Slider(1, 6, value=3, step=0.5, label="Guidance scale")
+            cfg_scale = gr.Slider(1, 10, value=3, step=0.5, label="Guidance scale")
             batch_size = gr.Slider(1, 6, value=1, step=1, label="Batch size")
         with gr.Row():
-            height = gr.Slider(label="Height", value=1024, minimum=1, maximum=4096, step=32)
-            width = gr.Slider(label="Width", value=1024, minimum=1, maximum=4096, step=32)
+            height = gr.Slider(label="Height", value=1024, minimum=8, maximum=2560, step=8)
+            width = gr.Slider(label="Width", value=1024, minimum=8, maximum=2560, step=8)
         with gr.Row():
             pixels = gr.Number(label="Pixel Ratio", value=1, interactive=False)
             seed = gr.Number(label="Seed", value=-1, minimum=-1, precision=0)
-        gallery = gr.Gallery(label="Generated images", height=800)
+        gallery = gr.Gallery(label="Generated images")
         ex = gr.Examples(examples=examples, inputs=[prompt, negative_prompt])
-    submit_btn.click(generate, [prompt, negative_prompt, samp_steps, guide_scale, batch_size, seed, height, width], [gallery], queue=True)
+    submit_btn.click(generate, [prompt, negative_prompt, samp_steps, cfg_scale, batch_size, seed, height, width], [gallery], queue=True)
     height.release(update_pixel_ratio, [height, width], [pixels, height], queue=False)
     width.release(update_pixel_ratio, [width, height], [pixels, width], queue=False)