Update handler.py

handler.py

@@ -173,8 +173,9 @@ def process_json_input(json_data):
         raise ValueError("Invalid JSON structure.")
 
 class EndpointHandler:
-    def __init__(self):
-        pass
+    def __init__(self, model_dir):
+        super().__init__()
+        self.model_dir = model_dir
 
     def __call__(self, data: Dict[str, Any]) -> Dict[str, Any]:
         try: