TexR6 commited on
Commit
6532771
β€’
1 Parent(s): 9efe5cc

removed encoder layer selection feature (for now)

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -109,7 +109,7 @@ def plot_weights(input_data, patch_weights):
109
  attn_map_img = attn_map_img.resize((224, 224), Image.ANTIALIAS)
110
  return attn_map_img
111
 
112
- attention_maps = plot_attention(random_image, "6")
113
 
114
  title_classify = "Image Based Plant Disease Identification πŸƒπŸ€“"
115
 
 
109
  attn_map_img = attn_map_img.resize((224, 224), Image.ANTIALIAS)
110
  return attn_map_img
111
 
112
+ attention_maps = plot_attention(random_image)
113
 
114
  title_classify = "Image Based Plant Disease Identification πŸƒπŸ€“"
115