update

README.md

@@ -1,5 +1,5 @@
 ---
-title: Task 1 CUB
+title: Visual CorrespondenceHuman Study - CUB
 emoji: 🚀
 colorFrom: green
 colorTo: gray
@@ -10,4 +10,4 @@ pinned: false
 license: mit
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces#reference
+Check out the configuration reference at <https://huggingface.co/docs/hub/spaces#reference>

app.py

@@ -33,13 +33,14 @@ classifier_predictions = {}
 selected_dataset = "CUB-iNAt-Unified"
 
 root_visualization_dir = "./visualizations/"
-viz_url = "https://static.taesiri.com/xai/CUB-iNAt-Unified.zip"
+viz_url = "https://drive.google.com/uc?id=1ifNYT2Jfnj61U1t144ZRE3c1PI_K89YA"
 viz_archivefile = "CUB-Final.zip"
 
-demonstration_url = "https://static.taesiri.com/xai/CUB-Demonstrations.zip"
+
+demonstration_url = "https://drive.google.com/uc?id=1YbAT-M_ZdjLlPuPPAaXbAOqYOqhAIJWs"
 demonst_zipfile = "demonstrations.zip"
 
-picklefile_url = "https://static.taesiri.com/xai/Task1-CUB-ALL.pickle"
+picklefile_url = "https://drive.google.com/uc?id=175CO7kfqedrmqO-3wXCN9gFFjnVhXFn8"
 prediction_root = "./predictions/"
 prediction_pickle = f"{prediction_root}predictions.pickle"
 
@@ -365,21 +366,17 @@ def main():
     global CLASSIFIER_TAG
 
     # Set the session state
-    # State Management and General Setup
     st.set_page_config(layout="wide")
-    st.title("TASK - 1 - CUB")
-
-    # st.write(classifier_predictions.keys())
-    # st.write(classifier_predictions["ILSVRC2012_val_00024646.JPEG"])
+    st.title("Visual CorrespondenceHuman Study - CUB")
 
     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(
@@ -400,7 +397,6 @@ def main():
             key="which_xai",
             index=default,
         )
-        # print(session_state.XAI_tool)
 
     if session_state.XAI_tool != "Unselected":
         st.markdown(f"## SELECTED METHOD ``{session_state.XAI_tool}``")
@@ -411,16 +407,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"