Update app.py

app.py

@@ -57,7 +57,7 @@ def generate(prompt, neg_prompt, samp_steps, cfg_scale, batch_size, seed, height
         width=width,
         generator=torch.manual_seed(seed),
     ).images
-    return [(img, f"Image {i+1}") for i, img in enumerate(images)], seed
+    return {gallery: gr.update(value = [(img, f"Image {i+1}") for i, img in enumerate(images)], height=height), seed}
         
 def set_base_model(base_model_id):
     vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
@@ -76,7 +76,7 @@ def set_base_model(base_model_id):
     return pipe
 
 def update_pixel_ratio(num1, num2):
-    return [round(num1*num2/1048576,3), num1-(num1%8)]
+    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',