Update backend/utils.py

backend/utils.py

@@ -17,14 +17,15 @@ from tqdm import trange
 import torch
 from transformers import AutoFeatureExtractor, AutoModelForImageClassification
 
-# @st.cache(allow_output_mutation=True)
-@st.cache_resource
+@st.cache(allow_output_mutation=True)
+# @st.cache_resource
 def load_dataset(data_index):
     with open(f'./data/preprocessed_image_net/val_data_{data_index}.pkl', 'rb') as file:
         dataset = pickle.load(file)
     return dataset
 
-@st.cache_resource
+@st.cache(allow_output_mutation=True)
+# @st.cache_resource
 def load_dataset_dict():
     dataset_dict = {}
     progress_empty = st.empty()
@@ -39,13 +40,15 @@ def load_dataset_dict():
     return dataset_dict
 
 
-@st.cache_data
+# @st.cache_data
+@st.cache(allow_output_mutation=True)
 def load_image(image_id):
     dataset = load_dataset(image_id//10000)
     image = dataset[image_id%10000]
     return image
 
-@st.cache_data
+# @st.cache_data
+@st.cache(allow_output_mutation=True)
 def load_images(image_ids):
     images = []
     for image_id in image_ids:
@@ -54,8 +57,8 @@ def load_images(image_ids):
     return images
 
 
-# @st.cache(allow_output_mutation=True, suppress_st_warning=True, show_spinner=False)
-@st.cache_resource
+@st.cache(allow_output_mutation=True, suppress_st_warning=True, show_spinner=False)
+# @st.cache_resource
 def load_model(model_name):
     with st.spinner(f"Loading {model_name} model! This process might take 1-2 minutes..."):
         if model_name == 'ResNet':
@@ -356,21 +359,21 @@ def _set_block_container_style(
     )
 
 
-@st.cache
-def get_dataframe() -> pd.DataFrame():
-    """Dummy DataFrame"""
-    data = [
-        {"quantity": 1, "price": 2},
-        {"quantity": 3, "price": 5},
-        {"quantity": 4, "price": 8},
-    ]
-    return pd.DataFrame(data)
+# @st.cache
+# def get_dataframe() -> pd.DataFrame():
+#     """Dummy DataFrame"""
+#     data = [
+#         {"quantity": 1, "price": 2},
+#         {"quantity": 3, "price": 5},
+#         {"quantity": 4, "price": 8},
+#     ]
+#     return pd.DataFrame(data)
 
 
-def get_plotly_fig():
-    """Dummy Plotly Plot"""
-    return px.line(data_frame=get_dataframe(), x="quantity", y="price")
+# def get_plotly_fig():
+#     """Dummy Plotly Plot"""
+#     return px.line(data_frame=get_dataframe(), x="quantity", y="price")
 
 
-def get_matplotlib_plt():
-    get_dataframe().plot(kind="line", x="quantity", y="price", figsize=(5, 3))
+# def get_matplotlib_plt():
+#     get_dataframe().plot(kind="line", x="quantity", y="price", figsize=(5, 3))