English
detection
open-world
open-set
Inference Endpoints
kelvinou01 commited on
Commit
a740a6e
1 Parent(s): 2299fd1
Files changed (1) hide show
  1. 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 = "/repository"
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(HOME, "weights", "groundingdino_swint_ogc.pth")
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):