Update handler.py

handler.py

@@ -18,8 +18,8 @@ from coresugg import ConversationPayload as ConversationPayloadSugg, create_conv
 from corechat import ConversationPayload as ConversationPayloadChat, get_conversation_suggestions, NUMBER_OF_MESSAGES_FOR_CONTEXT as NUMBER_OF_MESSAGES_FOR_CONTEXT_CHAT
 
 class EndpointHandler:
-    def __init__(self):
-        pass
+    def __init__(self, model_dir):
+        self.model_dir = model_dir
 
     def integration(self, data: Dict[str, Any]) -> Dict[str, Any]:
         payload = ConversationPayloadSugg(**data)