Update main.py

main.py

@@ -1,3 +1,4 @@
+from asyncio import sleep
 from typing import Union
 from fastapi import FastAPI
 from fastapi.encoders import jsonable_encoder
@@ -68,6 +69,7 @@ def index():
 @app.websocket("/accelerate")
 async def answer(ws: WebSocket):
   await accelerator.connect(ws)
+  while accelerator.connected(): sleep(10)
 
 @app.post("/map")
 def map(query: Union[str, None], items: Union[list[str], None]):