Update app.py

app.py

@@ -139,13 +139,19 @@ def predict_beetle(img):
 
 
 # gradio app
-with gr.Blocks() as demo:
+css = """
+button {
+    width: auto;  /* Set your desired width */
+}
+"""
+
+with gr.Blocks(css=css) as demo:
     with gr.Column(variant="panel"):
         with gr.Row(variant="compact"):
             inputs = gr.Image()
             # Use the `full_width` parameter directly
-            btn = gr.Button("Classify", full_width=False)
+            btn = gr.Button("Classify")
 
         # Set the gallery layout and height directly in the constructor
-        gallery = gr.Gallery(label="Show images", show_label=True, elem_id="gallery", layout="grid", cell_size=8, height="auto")
+        gallery = gr.Gallery(label="Show images", show_label=True, elem_id="gallery", columns=8, height="auto")
 demo.launch()