Update app.py

remove sos and eos tokens

app.py

@@ -28,6 +28,10 @@ def inference(prompt):
     net_attn_maps = get_net_attn_map(image.size)
     net_attn_maps = resize_net_attn_map(net_attn_maps, image.size)
     net_attn_maps = return_net_attn_map(net_attn_maps, pipe.tokenizer, prompt)
+
+    # remove sos and eos
+    net_attn_maps = [attn_map for attn_map in net_attn_maps if attn_map[1].split('_')[-1] != "<<|startoftext|>>"]
+    net_attn_maps = [attn_map for attn_map in net_attn_maps if attn_map[1].split('_')[-1] != "<<|endoftext|>>"]
     
     return image, net_attn_maps