Update app.py

app.py

@@ -35,7 +35,7 @@ css = """
 }
 """
 
-def generate(prompt, samp_steps, batch_size, seed, height, width, progress=gr.Progress(track_tqdm=True)):
+def generate(prompt, samp_steps, batch_size, seed, progress=gr.Progress(track_tqdm=True)):
     prompt = prompt.lower()
     if nsfw_filter:
         if prompt[:10] == "krebzonide":
@@ -52,8 +52,6 @@ def generate(prompt, samp_steps, batch_size, seed, height, width, progress=gr.Pr
         num_inference_steps=samp_steps,
         num_images_per_prompt=batch_size,
         guidance_scale=0.0,
-        height=height,
-        width=width,
         generator=torch.manual_seed(seed),
     ).images
     return gr.update(value = [(img, f"Image {i+1}") for i, img in enumerate(images)], height=height+90), seed
@@ -71,9 +69,6 @@ def set_base_model(base_model_id):
     pipe.to("cuda")
     return pipe
 
-def update_pixel_ratio(num1, num2):
-    return [round((num1-(num1%8))*num2/1048576,3), num1-(num1%8)]
-
 examples = [
     ['A group of 4 students from University of Wisconsin Stout sitting at a table talking, men and women, detailed faces, focused',
     'glitch, deformed, cross-eyed'],
@@ -98,21 +93,14 @@ with gr.Blocks(css=css) as demo:
         prompt = gr.Textbox(label="Prompt")
         submit_btn = gr.Button("Generate", elem_classes="btn-green")
         with gr.Row():
-            samp_steps = gr.Slider(1, 30, value=20, step=1, label="Sampling steps")
+            samp_steps = gr.Slider(1, 30, value=10, step=1, label="Sampling steps")
             batch_size = gr.Slider(1, 6, value=1, step=1, label="Batch size", interactive=True)
-        with gr.Row():
-            height = gr.Slider(label="Height", value=1024, minimum=8, maximum=1536, step=8)
-            width = gr.Slider(label="Width", value=1024, minimum=8, maximum=1024, 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(show_label=False, preview=True, container=False)
         with gr.Row():
             lastSeed = gr.Number(label="Last Seed", value=-1, interactive=False)
         ex = gr.Examples(examples=examples, inputs=[prompt, negative_prompt])
     submit_btn.click(generate, [prompt, samp_steps, batch_size, seed, height, width], [gallery, lastSeed], queue=True)
-    height.release(update_pixel_ratio, [height, width], [pixels, height], queue=False)
-    width.release(update_pixel_ratio, [width, height], [pixels, width], queue=False)
-
+    
 pipe = set_base_model(model_id)
 demo.launch(debug=True)