Update server.py

server.py

@@ -6,6 +6,12 @@ class EmbeddingService:
         self.keys = self.get_keys()
         self.conf = conf
 
+    def __enter__(self):
+        print("Start Embedding Service")
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        print("Exiting Embedding Service")
 
     def get_keys(self):
         "Get keys required to set up loading"