Update app.py

app.py

@@ -102,7 +102,7 @@ def process_image(model, tokenizer, transform, device, check_type, image, text):
     attn_map = similarity.reshape(len(text_embeds), resized_size[0], resized_size[1])
     # attn_map = similarity.reshape(len(text_embeds), *target_ratio)
     all_bpe_strings = [tokenizer.decode(input_id) for input_id in input_ids]
-    current_vis = generate_similiarity_map([image], attn_map, 
+    current_vis = generate_similiarity_map(images, attn_map, 
                                            [tokenizer.decode([i]) for i in input_ids], 
                                            [], target_ratio, src_size)