fix typo

app.py

@@ -155,8 +155,8 @@ with gr.Blocks(css=css) as demo:
                     inputs = [input_image]
                 )
             with gr.Column():
-                result = gr.Image(label="Result", columns=3)
-                heatmaps = gr.Gallery(label="Heatmap")
+                result = gr.Image(label="Result")
+                heatmaps = gr.Gallery(label="Heatmap", columns=3)
 
     submit_button.click(
         fn = main,