add options to turn features off

app.py

@@ -200,7 +200,7 @@ with st.sidebar:
         if use_only_relevance_2:
             # remove all qrels that are not relevance 2
             for qid, doc_rels in qrels.items():
-                qrels[qid] = {docid: rel for docid, rel in doc_rels.items() if rel == "2"}
+                qrels[qid] = {docid: rel for docid, rel in doc_rels.items() if rel == 2}
             # remove all queries that have no qrels
             queries = {qid: text for qid, text in queries.items() if qid in qrels}