fixed DEFAULT_LAYER value in app.py

app.py

@@ -115,7 +115,8 @@ def plot_weights(input_data, patch_weights, num_attention_block):
     attn_map_img = attn_map_img.resize((224, 224), Image.ANTIALIAS)
     attn_map_img.save(f"storage/attention_map_{num_attention_block}.png", "PNG")
 
-attention_maps = plot_attention(random_image, 6)
+DEFAULT_LAYER = "6"
+attention_maps = plot_attention(random_image, DEFAULT_LAYER)
 
 title_classify = "Image Based Plant Disease Identification 🍃🤓"
 
@@ -146,7 +147,7 @@ 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=["6"], label="Attention Layer")],
+                        value=DEFAULT_LAYER, label="Attention Layer")],
     outputs=gr.Gallery(value=attention_maps, label="Attention Maps").style(grid=(3, 4)),
     examples=example_list,
     title=title_attention,