add building model info
Browse files- model/inference_cpu.py +1 -0
model/inference_cpu.py
CHANGED
@@ -119,6 +119,7 @@ def zoom_in_zoom_out(args, segvol_model, image, image_resize, text_prompt, point
|
|
119 |
@st.cache_resource
|
120 |
def build_model():
|
121 |
# build model
|
|
|
122 |
clip_ckpt = 'model/config/clip'
|
123 |
resume = 'SegVol_v1.pth'
|
124 |
sam_model = sam_model_registry['vit']()
|
|
|
119 |
@st.cache_resource
|
120 |
def build_model():
|
121 |
# build model
|
122 |
+
st.write('building model')
|
123 |
clip_ckpt = 'model/config/clip'
|
124 |
resume = 'SegVol_v1.pth'
|
125 |
sam_model = sam_model_registry['vit']()
|