changed to uvicorn setup for HF v8

main/main.py

@@ -39,6 +39,9 @@ def create_app():
     # Initialize API with config
     api = InferenceApi(config)
 
+    # Initialize router with API instance
+    init_router(api)
+
     # Create LitServer instance
     server = ls.LitServer(
         api,
@@ -69,11 +72,6 @@ def create_app():
     # Set the response queue ID for the app
     app.response_queue_id = 0  # Since we're using a single worker
 
-    @app.on_event("startup")
-    async def startup_event():
-        """Initialize router on startup"""
-        await init_router(api)
-
     return app
 
 # Create the app instance for uvicorn

main/routes.py

@@ -19,7 +19,7 @@ router = APIRouter()
 logger = logging.getLogger(__name__)
 api = None
 
-async def init_router(inference_api: InferenceApi):
+def init_router(inference_api: InferenceApi):
     """Initialize router with an already setup API instance"""
     global api
     api = inference_api