Spaces:
Runtime error
Runtime error
Comment cache
Browse files
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.
|
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"}
|