From c4a3da109a01a3580428faa58e7ffaa31cb10955 Mon Sep 17 00:00:00 2001 From: wiktorlazarski Date: Wed, 23 Aug 2023 17:37:48 +0200 Subject: [PATCH] Upload the ResNet model again. --- apps/web_checking.py | 4 ++-- head_segmentation/constants.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) 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