Update app.py

app.py

@@ -53,7 +53,11 @@ grid_sizes = {
     "2x2": (2, 2),
     "2x3": (2, 3),
     "3x2": (3, 2),
-    "1x1": (1, 1)
+    "1x1": (1, 1),
+    "2x4": (2, 4),
+    "4x2": (4, 2),
+    "3x3": (3, 3),
+    "4x4": (4, 4)
 }
 
 @spaces.GPU(duration=60)
@@ -152,7 +156,7 @@ with gr.Blocks(css=css, theme="prithivMLmods/Minecraft-Theme") as demo:
 
         with gr.Row(visible=True):
             grid_size_selection = gr.Dropdown(
-                choices=["2x1", "1x2", "2x2", "2x3", "3x2", "1x1"],
+                choices=["2x1", "1x2", "2x2", "2x3", "3x2", "1x1", "2x4", "4x2", "3x3", "4x4"]
                 value="1x1",
                 label="Grid Size"
             )