Update app.py

app.py

@@ -141,9 +141,24 @@ with gr.Blocks() as app:
     api_input = gr.Textbox(label="Roboflow API Key", type="password")
     url_input = gr.Textbox(label="Roboflow Dataset URL (Segmentation)")
     run_btn = gr.Button("Convert")
-    before_gallery = gr.Gallery(label="Before (Segmentation)").style(grid=[5], height="auto")
-    after_gallery = gr.Gallery(label="After (Bounding Boxes)").style(grid=[5], height="auto")
-    run_btn.click(convert_seg_to_bbox, inputs=[api_input, url_input], outputs=[before_gallery, after_gallery])
+
+    before_gallery = gr.Gallery(
+        label="Before (Segmentation)",
+        columns=5,
+        height="auto"
+    )
+
+    after_gallery = gr.Gallery(
+        label="After (Bounding Boxes)",
+        columns=5,
+        height="auto"
+    )
+
+    run_btn.click(
+        convert_seg_to_bbox,
+        inputs=[api_input, url_input],
+        outputs=[before_gallery, after_gallery]
+    )
 
 if __name__ == "__main__":
     app.launch()