Skip to content

Commit 5dd84ef

Browse files
authored
Added huggingface_hub and remove onnx (#87)
1 parent ddcee01 commit 5dd84ef

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

requirements.txt

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,8 +2,7 @@ segment-anything-py
22
opencv-python
33
pycocotools
44
matplotlib
5-
onnxruntime; python_version < '3.11'
6-
onnx
5+
huggingface_hub
76
geopandas
87
rasterio
98
tqdm

samgeo/text_sam.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
from PIL import Image
1212
from segment_anything import sam_model_registry
1313
from segment_anything import SamPredictor
14+
from huggingface_hub import hf_hub_download
1415
from .common import *
1516

1617
try:
@@ -29,7 +30,7 @@
2930
from groundingdino.util.inference import predict
3031
from groundingdino.util.slconfig import SLConfig
3132
from groundingdino.util.utils import clean_state_dict
32-
from huggingface_hub import hf_hub_download
33+
3334
except ImportError:
3435
print("Installing GroundingDINO...")
3536
install_package("groundingdino-py")

0 commit comments

Comments
 (0)