kelvinou01
commited on
Commit
•
46c271a
1
Parent(s):
7a1d722
Add param to handler
Browse files- handler.py +1 -1
handler.py
CHANGED
@@ -8,7 +8,7 @@ CONFIG_PATH = os.path.join(HOME, "GroundingDINO/groundingdino/config/GroundingDI
|
|
8 |
WEIGHTS_PATH = os.path.join(HOME, "weights", "groundingdino_swint_ogc.pth")
|
9 |
|
10 |
class EndpointHandler():
|
11 |
-
def __init__(self):
|
12 |
# Preload all the elements you are going to need at inference.
|
13 |
self.model = load_model(CONFIG_PATH, WEIGHTS_PATH)
|
14 |
|
|
|
8 |
WEIGHTS_PATH = os.path.join(HOME, "weights", "groundingdino_swint_ogc.pth")
|
9 |
|
10 |
class EndpointHandler():
|
11 |
+
def __init__(self, path):
|
12 |
# Preload all the elements you are going to need at inference.
|
13 |
self.model = load_model(CONFIG_PATH, WEIGHTS_PATH)
|
14 |
|