d

handler.py

@@ -17,14 +17,8 @@ WEIGHTS_PATH = os.path.join("model", "weights", "groundingdino_swint_ogc.pth")
 class EndpointHandler():
     def __init__(self, path):
         # Preload all the elements you are going to need at inference.
-        print("===printing cwd===")
-        for root, dirs, files in os.walk(os.getcwd()):
-            print(f"Files in {root}:")
-            for file in files:
-                print(os.path.join(root, file))
-        print("===done printing cwd===")
-
-        self.model = load_model(CONFIG_PATH, WEIGHTS_PATH)
+
+        self.model = load_model(CONFIG_PATH, path)
 
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
         """