Update app.py

app.py

@@ -115,17 +115,17 @@ def detect_objects(
         display_mask=display_mask,
         mask=img_input_mask
     )
-    # decoder_attention_map_img = visualize_attention_map(
-    #     image_input, decoder_attention_map
-    # )
-    # encoder_attention_map_img = visualize_attention_map(
-    #     image_input, encoder_attention_map
-    # )
+    decoder_attention_map_img = visualize_attention_map(
+        image_input, decoder_attention_map
+    )
+    encoder_attention_map_img = visualize_attention_map(
+        image_input, encoder_attention_map
+    )
 
     return (
         viz_img,
-        # decoder_attention_map_img,
-        # encoder_attention_map_img,
+        decoder_attention_map_img,
+        encoder_attention_map_img,
     )
 
 
@@ -242,16 +242,16 @@ with gr.Blocks(css=css) as app:
                         with gr.Column():
                             img_output_from_upload = gr.Image(shape=(900, 900))
 
-        # with gr.TabItem("Attentions visualization"):
-        #     gr.Markdown("""Encoder attentions""")
-        #     with gr.Row():
-        #         encoder_att_map_output = gr.Image(shape=(850, 850))
-        #     gr.Markdown("""Decoder attentions""")
-        #     with gr.Row():
-        #         decoder_att_map_output = gr.Image(shape=(850, 850))
-        # with gr.TabItem("Dataset details"):
-        #     with gr.Row():
-        #         gr.Markdown(description)
+        with gr.TabItem("Attentions visualization"):
+            gr.Markdown("""Encoder attentions""")
+            with gr.Row():
+                encoder_att_map_output = gr.Image(shape=(850, 850))
+            gr.Markdown("""Decoder attentions""")
+            with gr.Row():
+                decoder_att_map_output = gr.Image(shape=(850, 850))
+        with gr.TabItem("Dataset details"):
+            with gr.Row():
+                gr.Markdown(description)
 
     detect_button.click(
         detect_objects,
@@ -271,8 +271,8 @@ with gr.Blocks(css=css) as app:
         ],
         outputs=[
             img_output_from_upload,
-            #decoder_att_map_output,
-            #encoder_att_map_output,
+            decoder_att_map_output,
+            encoder_att_map_output,
         ],
         queue=True,
     )