wondervictor commited on
Commit
60f2f84
·
verified ·
1 Parent(s): f021c77

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -116,8 +116,8 @@ def inference_point(input_img, img_state,):
116
 
117
  demo = SAMPointVisualizationDemo(cfg, 0.8, sam2_model, clip_model,mask_adapter)
118
 
119
-
120
- _, visualized_output = demo.run_on_image_with_points(img_state.img, points)
121
  return visualized_output
122
 
123
 
 
116
 
117
  demo = SAMPointVisualizationDemo(cfg, 0.8, sam2_model, clip_model,mask_adapter)
118
 
119
+ text_features = torch.from_numpy(np.load("./text_embedding/lvis_coco_text_embedding.npy")).cuda()
120
+ _, visualized_output = demo.run_on_image_with_points(img_state.img, points,text_features)
121
  return visualized_output
122
 
123