Spaces:
Runtime error
Runtime error
Update grounding_model.py
Browse files- grounding_model.py +3 -1
grounding_model.py
CHANGED
|
@@ -11,7 +11,9 @@ from PIL import ImageDraw, ImageFont
|
|
| 11 |
|
| 12 |
from constants.constant import DARKER_COLOR_MAP, LIGHTER_COLOR_MAP, COLORS
|
| 13 |
import groundingdino.datasets.transforms as T
|
| 14 |
-
from groundingdino import build_groundingdino
|
|
|
|
|
|
|
| 15 |
from groundingdino.util.inference import annotate, load_image, predict
|
| 16 |
from groundingdino.util.utils import clean_state_dict
|
| 17 |
|
|
|
|
| 11 |
|
| 12 |
from constants.constant import DARKER_COLOR_MAP, LIGHTER_COLOR_MAP, COLORS
|
| 13 |
import groundingdino.datasets.transforms as T
|
| 14 |
+
# from groundingdino import build_groundingdino
|
| 15 |
+
|
| 16 |
+
from groundingdino.models.GroundingDINO.groundingdino import build_groundingdino
|
| 17 |
from groundingdino.util.inference import annotate, load_image, predict
|
| 18 |
from groundingdino.util.utils import clean_state_dict
|
| 19 |
|