Update app.py

app.py

@@ -327,8 +327,8 @@ if __name__ == "__main__":
 
             with gr.Column():
                 gallery = gr.Gallery(
-                    label="Generated images", show_label=False, elem_id="gallery"
-                ).style(preview=True, grid=2, object_fit="scale-down")
+                    label="Generated images", show_label=False, elem_id="gallery",
+                    preview=True, object_fit="scale-down")
 
         run_button.click(fn=run_grounded_sam, inputs=[
             input_image, text_prompt, task_type, inpaint_prompt, box_threshold, text_threshold, iou_threshold, inpaint_mode], outputs=gallery)