fuck this shit!

app.py

@@ -81,8 +81,9 @@ def predict_tensor(img_tensor):
 
 random_image = PIL.Image.open('examples/TomatoYellowCurlVirus6.JPG').convert('RGB')
 
-def plot_attention(image, encoder_layer_num):
+def plot_attention(image, encoder_layer_num=5):
     """Given an input image, plot the average attention weight given to each image patch by each attention head."""
+    print(f'Selected encoder layer num: {encoder_layer_num}')
     attention_map_outputs = []
     input_data = data_transforms(image)
     with nopdb.capture_call(vision_transformer.blocks[encoder_layer_num].attn.forward) as attn_call:
@@ -139,7 +140,7 @@ classify_interface = gr.Interface(
 attention_interface = gr.Interface(
     fn=plot_attention,
     inputs=[gr.Image(type="pil", label="Image"),
-           gr.Dropdown(choices=["1", "2", "3", "4", "5", "6", "7", "8", "9", "10", "11", "12"], value=5, type="index")],
+           gr.Dropdown(choices=["1", "2", "3", "4", "5", "6", "7", "8", "9", "10", "11", "12"], value="6", type="index")],
     outputs=gr.Gallery(value=attention_maps, label="Attention Maps").style(grid=(3, 4)),
     examples=example_list,
     title=title_attention,