Update app.py

app.py

@@ -483,7 +483,7 @@ if check_valid_args(run1_file, run2_file, dataset_name, qrels, queries, corpus):
                 container_two_docs_rel = st.container()
                 col_run1, col_run2 = container_two_docs_rel.columns(2, gap="medium")
                 relevant_docs = list(qrels[str(inst_num)].keys())[:n_relevant_docs]
-                relevant_score = {ind_doc_id: item[str(inst_num)][ind_doc_id] for ind_doc_id in relevant_docs}
+                relevant_score = {ind_doc_id: qrels[str(inst_num)][ind_doc_id] for ind_doc_id in relevant_docs}
                 doc_texts = [(doc_id, corpus[doc_id]["title"] if "title" in corpus[doc_id] else "", corpus[doc_id]["text"], relevance_score[doc_id]) for doc_id in relevant_docs]
 
                 if doc_expansion1 is not None and run1_uses_doc_expansion != "None":