update

app.py

@@ -341,19 +341,16 @@ def main():
     # Set the session state
     # State Management and General Setup
     st.set_page_config(layout="wide")
-    st.title("TASK - 1 - ImageNetREAL")
-
-    # st.write(classifier_predictions.keys())
-    # st.write(classifier_predictions["ILSVRC2012_val_00024646.JPEG"])
+    st.title("Visual CorrespondenceHuman Study - ImageNet")
 
     options = [
         "Unselected",
         "NOXAI",
         "KNN",
-        "EMD Nearest Neighbors",
-        "EMD Correspondence",
-        "CHM Nearest Neighbors",
-        "CHM Correspondence",
+        "EMD-Corr Nearest Neighbors",
+        "EMD-Corr Correspondence",
+        "CHM-Corr Nearest Neighbors",
+        "CHM-Corr Correspondence",
     ]
 
     st.markdown(
@@ -385,16 +382,16 @@ def main():
         elif session_state.XAI_tool == "KNN":
             selected_xai_tool = load_knn_nns
             CLASSIFIER_TAG = "knn"
-        elif session_state.XAI_tool == "CHM Nearest Neighbors":
+        elif session_state.XAI_tool == "CHM-Corr Nearest Neighbors":
             selected_xai_tool = load_chm_nns
             CLASSIFIER_TAG = "CHM"
-        elif session_state.XAI_tool == "CHM Correspondence":
+        elif session_state.XAI_tool == "CHM-Corr Correspondence":
             selected_xai_tool = load_chm_corrs
             CLASSIFIER_TAG = "CHM"
-        elif session_state.XAI_tool == "EMD Nearest Neighbors":
+        elif session_state.XAI_tool == "EMD-Corr Nearest Neighbors":
             selected_xai_tool = load_emd_nns
             CLASSIFIER_TAG = "EMD"
-        elif session_state.XAI_tool == "EMD Correspondence":
+        elif session_state.XAI_tool == "EMD-Corr Correspondence":
             selected_xai_tool = load_emd_corrs
             CLASSIFIER_TAG = "EMD"