Update app.py

app.py

@@ -97,9 +97,9 @@ with gr.Blocks() as interface:
     with gr.Row():
         for card, url in valid_card_images.items():
             with gr.Column():
-                # Combine the image and button
-                gr.Image(value=url, label=card, interactive=False).style(width=70, height=70)  # Adjust image size
-                gr.Button(card).click(
+                # Combine the image and button in a smaller size
+                gr.Image(value=url, interactive=False, elem_id=f"image-{card}")  # Image display
+                gr.Button(card, elem_id=f"button-{card}").click(
                     toggle_card,
                     inputs=[gr.Textbox(card), selected_deck],
                     outputs=[selected_deck],
@@ -115,4 +115,13 @@ with gr.Blocks() as interface:
     predict_button = gr.Button("Make Prediction")
     predict_button.click(validate_and_predict, inputs=[selected_deck], outputs=[result])
 
+# Add custom CSS to adjust the image size
+interface.css = """
+img {
+    width: 70px !important;
+    height: 70px !important;
+    margin-bottom: 5px;
+}
+"""
+
 interface.launch()