added some logs

main.py

@@ -1,4 +1,5 @@
 import uuid
+import logging
 import asyncio
 import transformers
 from fastapi import FastAPI
@@ -27,6 +28,15 @@ MAX_BATCH_SIZE = 128
 query_queue: asyncio.Queue = asyncio.Queue()
 results: dict[str, dict] = {}
 classifier = None  # will be initialized in lifespan
+logging.basicConfig(
+    level=logging.INFO,
+    format="%(asctime)s - %(levelname)s - %(message)s",
+)
+logger = logging.getLogger("uvicorn")
+logger.info("Starting the application...")
+logger.info(f"Using model: {MODEL_NAME}")
+logger.info(f"Batch process interval: {BATCH_PROCESS_INTERVAL}")
+logger.info(f"Max batch size: {MAX_BATCH_SIZE}")
 
 
 # ----------------------------- #
@@ -87,9 +97,12 @@ async def lifespan(_: FastAPI):
     global classifier
     classifier = load_classifier(MODEL_NAME)
     _ = classifier("Startup warm-up sentence.")
+    logger.info("Model loaded successfully.")
     queue_task = asyncio.create_task(process_queue())
     yield
     queue_task.cancel()
+    logger.info("Shutting down the application...")
+    logger.info("Model unloaded successfully.")
     try:
         await queue_task
     except asyncio.CancelledError:
@@ -107,6 +120,7 @@ app = FastAPI(lifespan=lifespan)
 # ----------------------------- #
 @app.post("/classify")
 async def classify(query: Query):
+    logger.info(f"Received query: {query.sentence}")
     query_id = str(uuid.uuid4())
     await query_queue.put({"id": query_id, "sentence": query.sentence})