Update app.py

app.py

@@ -187,9 +187,9 @@ def generate_random_seed():
     return random.randint(0, 2**32 - 1)
 
 MODEL_SAMPLE_LIMITS = {
-    "S": {"min":1, "max": 16, "default": 4},
-    "B": {"min":1, "max": 12, "default": 3},
-    "L": {"min":1, "max": 4, "default": 1}
+    "S": {"min":1, "max": 18, "default": 6},
+    "B": {"min":1, "max": 12, "default": 4},
+    "L": {"min":1, "max": 6, "default": 2}
 }
 
 def update_sample_slider(dit_size):
@@ -281,8 +281,8 @@ with gr.Blocks(title="DiT Diffusion Model Generator", theme=gr.themes.Soft()) as
         with gr.Column(scale=2):
             output_gallery = gr.Gallery(
                 label="Generated Images",
-                columns=4,
-                rows=3,
+                columns=3,
+                rows=2,
                 height=550,
                 object_fit="contain",
                 allow_preview=True,