English
detection
open-world
open-set
Inference Endpoints
kelvinou01 commited on
Commit
46c271a
1 Parent(s): 7a1d722

Add param to handler

Browse files
Files changed (1) hide show
  1. 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