Update app.py

app.py

@@ -111,15 +111,14 @@ with gr.Blocks(css=css) as app:
                     img_output_from_upload = gr.Image(shape=(850, 850))
         with gr.TabItem("Attention maps visualization"):
             with gr.Row():
-                with gr.Column():
-                    gr.Markdown("""Encoder attentions""")
-                    encoder_att_map_output = gr.Image(shape=(850, 850))
-                with gr.Column():
-                    gr.Markdown("""Decoder attentions""")
-                    decoder_att_map_output = gr.Image(shape=(850, 850))
-                with gr.Column():
-                    gr.Markdown("""Cross attentions""")
-                    cross_att_map_output = gr.Image(shape=(850, 850))
+                gr.Markdown("""Encoder attentions""")
+                encoder_att_map_output = gr.Image(shape=(850, 850))
+            with gr.Row():
+                gr.Markdown("""Decoder attentions""")
+                decoder_att_map_output = gr.Image(shape=(850, 850))
+            with gr.Row():
+                gr.Markdown("""Cross attentions""")
+                cross_att_map_output = gr.Image(shape=(850, 850))
         with gr.TabItem("Model details"):
             with gr.Row():
                 model_details = gr.Markdown(""" """)