Update app.py

app.py

@@ -120,12 +120,10 @@ with gr.Blocks() as app:
             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)?')
                 top_class_btn = gr.Button("Submit", variant='primary')
                 tc_clear_btn = gr.ClearButton()
 
-    with gr.Row():
-        if_show_grad_cam = gr.Checkbox(value=True, label='Show Class Activation Map (What the model sees)?')
-
     with gr.Row(visible=True) as top_class_output:
         with gr.Column(visible=True)  as top_class_output:
             top_class_output_img = gr.Image(interactive=False, label='Prediction Output')