Update pages/2_SmoothGrad.py

pages/2_SmoothGrad.py

@@ -104,8 +104,8 @@ grids = make_grid(cols=2+len(selected_models)*2, rows=len(image_ids)+1)
 #     models[model_name], feature_extractors[model_name] = load_model(model_name)
 
 
-# @st.cache(allow_output_mutation=True)
-@st.cache_data
+@st.cache(allow_output_mutation=True)
+# @st.cache_data
 def generate_images(image_id, model_name):
     j = image_ids.index(image_id)
     image = images[j]['image']