Update app.py

app.py

@@ -7,7 +7,6 @@ from datasets import Dataset
 from PIL import Image
 import io
 import uuid
-import shutil
 
 DATA_DIR = "/data"
 IMAGES_DIR = os.path.join(DATA_DIR, "images")
@@ -140,7 +139,7 @@ with gr.Blocks(theme="huggingface") as iface:
     dataset_info = gr.Textbox(label="Dataset Info", value=dataset_builder.get_dataset_info())
 
     gr.Markdown("## Dataset Preview")
-    preview_gallery = gr.Gallery(label="Recent Additions", show_label=False, elem_id="preview_gallery").style(grid=5, height="auto")
+    preview_gallery = gr.Gallery(label="Recent Additions", show_label=False, elem_id="preview_gallery", columns=5, rows=1, height="auto")
 
     add_button.click(add_image_to_dataset, inputs=url_input, outputs=[result_output, dataset_info, preview_gallery])
 
@@ -149,7 +148,7 @@ with gr.Blocks(theme="huggingface") as iface:
     create_hf_button.click(create_huggingface_dataset, inputs=[], outputs=hf_result)
 
     view_dataset_button = gr.Button("View Dataset")
-    dataset_gallery = gr.Gallery(label="Dataset Contents", show_label=False, elem_id="dataset_gallery").style(grid=5, height="auto")
+    dataset_gallery = gr.Gallery(label="Dataset Contents", show_label=False, elem_id="dataset_gallery", columns=5, rows=4, height="auto")
     view_dataset_button.click(view_dataset, inputs=[], outputs=dataset_gallery)
 
 # Launch the interface