Spaces:
Runtime error
Runtime error
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 |
|