Update app.py

app.py

@@ -68,6 +68,9 @@ def set_base_model(base_model_id):
     )
     pipe.to("cuda")
     return pipe
+
+def update_pixel_ratio(height, width):
+    return round(height*width/1048576,3)
     
 with gr.Blocks(css=css) as demo:
     with gr.Column():
@@ -79,11 +82,16 @@ with gr.Blocks(css=css) as demo:
             guide_scale = gr.Slider(1, 6, value=3, step=0.5, label="Guidance scale")
             batch_size = gr.Slider(1, 6, value=1, step=1, label="Batch size")
         with gr.Row():
-            seed = gr.Slider(label="Seed", value=-1, minimum=-1, maximum=999999, step=1)
             height = gr.Slider(label="Height", value=1024, minimum=512, maximum=2048, step=16)
             width = gr.Slider(label="Width", value=1024, minimum=512, maximum=2048, step=16)
+        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)
     submit_btn.click(generate, [prompt, negative_prompt, samp_steps, guide_scale, batch_size, seed, height, width], [gallery], queue=True)
+    height.change(update_pixel_ratio, [height, width], [pixels], queue=False)
+    width.change(update_pixel_ratio, [height, width], [pixels], queue=False)
+    
 
 pipe = set_base_model(model_id)
 demo.launch(debug=True)