yuxindu commited on
Commit
8de8404
·
1 Parent(s): 90d16cf

add building model info

Browse files
Files changed (1) hide show
  1. 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']()