Update app.py

app.py

@@ -32,7 +32,7 @@ def display_dict(data):
     all_context = data["all_context"]
     highlighted_context = ""
     for idx, sentence in enumerate(all_context):
-        if idx == data["perturb_sentence_id"]:
+        if idx in data["perturb_sentence_id"]:
             highlighted_context += f"<span style='color:red;'>{sentence}</span> "
         elif idx in data["target_context_ids"]:
             highlighted_context += f"**{sentence}** "