curt-park commited on
Commit
881cb83
·
1 Parent(s): 58b1022

Comment cache

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ from isegm.inference import clicker as ck
12
  from isegm.inference import utils
13
  from isegm.inference.predictors import get_predictor
14
 
15
- @st.cache
16
  def load_model(model_path, device):
17
  model = utils.load_is_model(model_path, device, cpu_dist_maps=True)
18
  predictor_params = {"brs_mode": "NoBRS"}
 
12
  from isegm.inference import utils
13
  from isegm.inference.predictors import get_predictor
14
 
15
+ # @st.cache_resource # doesn't in huggingface
16
  def load_model(model_path, device):
17
  model = utils.load_is_model(model_path, device, cpu_dist_maps=True)
18
  predictor_params = {"brs_mode": "NoBRS"}