Update app.py

app.py

@@ -208,14 +208,14 @@ with gr.Blocks(title="Trash Classification", css="#custom_header {min-height: 3r
 
     with gr.Column():
         with gr.Column():
-            with gr.Box():
+            with gr.Group():
                 gr.Markdown("## Inputs", elem_id="custom_header")
                 input_image = gr.Image(label="Input Image")
                 input_image.style(height=240)
                 btn = gr.Button(value="Submit")
                 btn.style(full_width=True)
         with gr.Column():
-            with gr.Box():
+            with gr.Group():
                 gr.Markdown("## Outputs", elem_id="custom_header")
                 recycling_qn = gr.outputs.Textbox(label="Is this recyclable?")
                 recycling_advice = gr.outputs.Textbox(label="It is not recyclable when:")