Update app.py

app.py

@@ -22,7 +22,6 @@ def make_prediction(img, feature_extractor, model):
         processed_outputs[0],
         outputs["decoder_attentions"],
         outputs["encoder_attentions"],
-        # outputs["cross_attentions"],
     )
 
 
@@ -37,7 +36,6 @@ def detect_objects(model_name, image_input, threshold, display_mask=False, img_i
         processed_outputs,
         decoder_attention_map,
         encoder_attention_map,
-        # cross_attention_map,
     ) = make_prediction(image_input, feature_extractor, model)
 
     viz_img = visualize_prediction(
@@ -54,13 +52,11 @@ def detect_objects(model_name, image_input, threshold, display_mask=False, img_i
     encoder_attention_map_img = visualize_attention_map(
         image_input, encoder_attention_map
     )
-    # cross_attention_map_img = visualize_attention_map(image_input, cross_attention_map)
 
     return (
         viz_img,
         decoder_attention_map_img,
         encoder_attention_map_img,
-        # cross_attention_map_img,
         model_details
     )
 
@@ -123,9 +119,6 @@ with gr.Blocks(css=css) as app:
             gr.Markdown("""Decoder attentions""")
             with gr.Row():
                 decoder_att_map_output = gr.Image(shape=(850, 850))
-            # gr.Markdown("""Cross attentions""")
-            # with gr.Row():
-            #    cross_att_map_output = gr.Image(shape=(850, 850))
         with gr.TabItem("Model details"):
             with gr.Row():
                 model_details = gr.Markdown(""" """)