Spaces:
Runtime error
Runtime error
Use streamlit's newer cache method
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 |
-
|
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.experimental_memo
|
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"}
|