Update app.py

app.py

@@ -149,7 +149,7 @@ with gr.Blocks(css=css) as demo:
                 input_image = gr.Image(label="Image Input", type="filepath")
                 submit_button = gr.Button("Submit")
             with gr.Column():
-                result = gr.Image(label="Result")
+                result = gr.Image(label="Result", columns=3)
                 heatmaps = gr.Gallery(label="Heatmap")
 
     submit_button.click(