Update handler.py
Browse files- handler.py +3 -2
handler.py
CHANGED
|
@@ -173,8 +173,9 @@ def process_json_input(json_data):
|
|
| 173 |
raise ValueError("Invalid JSON structure.")
|
| 174 |
|
| 175 |
class EndpointHandler:
|
| 176 |
-
def __init__(self):
|
| 177 |
-
|
|
|
|
| 178 |
|
| 179 |
def __call__(self, data: Dict[str, Any]) -> Dict[str, Any]:
|
| 180 |
try:
|
|
|
|
| 173 |
raise ValueError("Invalid JSON structure.")
|
| 174 |
|
| 175 |
class EndpointHandler:
|
| 176 |
+
def __init__(self, model_dir):
|
| 177 |
+
super().__init__()
|
| 178 |
+
self.model_dir = model_dir
|
| 179 |
|
| 180 |
def __call__(self, data: Dict[str, Any]) -> Dict[str, Any]:
|
| 181 |
try:
|