wakeup update

embed_utils.py

@@ -47,6 +47,7 @@ async def embed_wrapper(input_ds, temp_file):
 
 
 def wake_up_endpoint():
+    endpoint.fetch()
     if endpoint.status != 'running':
         logger.info("Starting up TEI endpoint")
         endpoint.resume().wait().fetch()
@@ -62,10 +63,3 @@ def wake_up_endpoint():
     logger.info("TEI endpoint is up")
 
     return
-
-    # endpoint.fetch()
-    # if endpoint.status != 'running':
-    #     logger.info("Starting up TEI endpoint")
-    #     endpoint.resume().wait()
-    # logger.info("TEI endpoint is up")
-    # return

main.py

@@ -109,11 +109,4 @@ def chunk_and_embed(input_ds_name):
     chunked_tmp_file = chunk(input_ds_name)
     embed(chunked_tmp_file)
 
-    return {"processed": True}
-
-
-# For debugging
-
-# import uvicorn
-# if __name__ == "__main__":
-#     uvicorn.run(app, host="0.0.0.0", port=7860)
+    return {"processed": True}