Merge pull request #29 from wiktorlazarski/update_link

Upload the ResNet model again.
wiktorlazarski · Aug 23, 2023 · cc16033 · cc16033
2 parents b3abb09 + c4a3da1
commit cc16033
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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