Update embed_utils.py

embed_utils.py

@@ -51,15 +51,15 @@ def wake_up_endpoint():
     if endpoint.status != 'running':
         logger.info("Starting up TEI endpoint")
         endpoint.resume().wait().fetch()
-        # n_loop = 0
-        # while requests.get(
-        #     url=endpoint.url,
-        #     headers={"Authorization": f"Bearer {env_config.hf_token}"}
-        # ).status_code != 200:
-        #     time.sleep(2)
-        #     n_loop += 1
-        #     if n_loop > 20:
-        #         raise TimeoutError("TEI endpoint is unavailable")
+        n_loop = 0
+        while requests.get(
+            url=endpoint.url,
+            headers={"Authorization": f"Bearer {env_config.hf_token}"}
+        ).status_code != 200:
+            time.sleep(2)
+            n_loop += 1
+            if n_loop > 20:
+                raise TimeoutError("TEI endpoint is unavailable")
     logger.info("TEI endpoint is up")
 
     return