Update app.py

app.py

@@ -113,10 +113,13 @@ with gr.Blocks() as app:
                                         value=[img[0] for img in sample_images], columns=7, rows=2)
 
         with gr.Column():
-            top_pred_image = gr.Image(label='Upload Image or Select from the gallery')
+            with gr.Row():
+                top_pred_image = gr.Image(label='Upload Image or Select from the gallery')
 
             with gr.Row():
                 if_show_grad_cam = gr.Checkbox(value=True, label='Show Class Activation Map (What the model sees)?')
+                
+            with gr.Row():
                 top_class_btn = gr.Button("Submit", variant='primary')
                 tc_clear_btn = gr.ClearButton()