Update app.py

app.py

@@ -143,11 +143,9 @@ with gr.Blocks() as demo:
     with gr.Column(variant="panel"):
         with gr.Row(variant="compact"):
             inputs = gr.Image()
-            btn = gr.Button("Classify")
+            # Use the `full_width` parameter directly
+            btn = gr.Button("Classify", full_width=False)
 
-        gallery = gr.Gallery(
-            label="Show images", show_label=True, elem_id="gallery"
-        )
-
-    btn.click(predict_beetle, inputs, gallery)
+        # 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")
 demo.launch()