Update app.py

app.py

@@ -86,10 +86,11 @@ def app():
                 gallery = gr.Gallery(
                     label="Image Gallery",
                     value=gallery_images,
-                    columns=3,
+                    columns=3,  # Set number of columns directly in the constructor
                     rows=2,
                     height="auto"
-                ).style(grid=3)
+                )
+
 
             with gr.Column():
                 output_image = gr.Image(label="Result Image", type="pil")