Update app.py

app.py

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