Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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 |
|