Update app.py

app.py

@@ -37,10 +37,6 @@ model_num_of_layers = {
     'Stable Diffusion 2.1': 22,
 }
 
-num_rows_dict = {
-    'Stable Diffusion 1.4': 2,
-    'Stable Diffusion 2.1': 4,
-}
 
 def generate_images(prompt, model, seed):
     seed = random.randint(0, MAX_SEED) if seed == -1 else seed
@@ -98,9 +94,9 @@ with gr.Blocks() as demo:
 
 
     generate_button = gr.Button("Generate Image")
-    num_rows = num_rows_dict[model]
+
     with gr.Column():
-        gallery = gr.Gallery(label="Generated Images", columns=6, rows=num_rows, object_fit="contain", height="auto")
+        gallery = gr.Gallery(label="Generated Images", columns=6, rows=1, object_fit="contain", height="auto")
 
     outputs = [gallery]