Update embed_utils.py

embed_utils.py

@@ -49,8 +49,6 @@ def wake_up_endpoint():
     endpoint.fetch()
     if endpoint.status != 'running':
         logger.info("Starting up TEI endpoint")
-        endpoint.resume()
-        endpoint.wait()
-        endpoint.fetch()
+        endpoint.resume().wait()
     logger.info("TEI endpoint is up")
     return