diff --git a/apps/web_checking.py b/apps/web_checking.py
index eaece44..f7097af 100644
--- a/apps/web_checking.py
+++ b/apps/web_checking.py
@@ -6,12 +6,12 @@
 import head_segmentation.visualization as vis
 
 
-@st.cache(allow_output_mutation=True)
+@st.cache_resource()
 def load_model() -> seg_pipeline.HumanHeadSegmentationPipeline:
     return seg_pipeline.HumanHeadSegmentationPipeline()
 
 
-@st.cache(allow_output_mutation=True)
+@st.cache_resource()
 def create_vis_module() -> vis.VisualizationModule:
     return vis.VisualizationModule()
 
diff --git a/head_segmentation/constants.py b/head_segmentation/constants.py
index ea3b47d..a107530 100644
--- a/head_segmentation/constants.py
+++ b/head_segmentation/constants.py
@@ -9,5 +9,5 @@
 
 # fmt: off
 HEAD_SEGMENTATION_MODEL_PATH = os.path.join(os.path.dirname(os.path.abspath(__file__)), "model", "head_segmentation.ckpt")
-HEAD_SEGMENTATION_MODEL_URL = "https://drive.google.com/uc?id=1Asm8JMFQwiHGGuQLXG_qnHvspWGdGuxw"
+HEAD_SEGMENTATION_MODEL_URL = "https://drive.google.com/uc?id=1RxXX4g3zMk2CwDtsq-gGleq7aSGoUUHf"
 # fmt: on