kelvinou01
commited on
Commit
•
a740a6e
1
Parent(s):
2299fd1
FIx
Browse files- handler.py +2 -2
handler.py
CHANGED
@@ -5,13 +5,13 @@ import groundingdino
|
|
5 |
from groundingdino.util.inference import load_model, load_image, predict, annotate
|
6 |
|
7 |
# /app
|
8 |
-
HOME =
|
9 |
|
10 |
# /opt/conda/lib/python3.9/site-packages/groundingdino
|
11 |
PACKAGE_HOME = os.path.dirname(groundingdino.__file__)
|
12 |
|
13 |
CONFIG_PATH = os.path.join(PACKAGE_HOME, "config", "GroundingDINO_SwinT_OGC.py")
|
14 |
-
WEIGHTS_PATH = os.path.join(
|
15 |
|
16 |
class EndpointHandler():
|
17 |
def __init__(self, path):
|
|
|
5 |
from groundingdino.util.inference import load_model, load_image, predict, annotate
|
6 |
|
7 |
# /app
|
8 |
+
HOME = os.getcwd()
|
9 |
|
10 |
# /opt/conda/lib/python3.9/site-packages/groundingdino
|
11 |
PACKAGE_HOME = os.path.dirname(groundingdino.__file__)
|
12 |
|
13 |
CONFIG_PATH = os.path.join(PACKAGE_HOME, "config", "GroundingDINO_SwinT_OGC.py")
|
14 |
+
WEIGHTS_PATH = os.path.join("repository", "weights", "groundingdino_swint_ogc.pth")
|
15 |
|
16 |
class EndpointHandler():
|
17 |
def __init__(self, path):
|