Update app.py

app.py

@@ -1,7 +1,7 @@
 import os
 
 # Set YOLO_CONFIG_DIR before any Ultralytics imports
-os.environ["YOLO_CONFIG_DIR"] = "/tmp"
+os.environ["YOLO_CONFIG_DIR"] = "/tmp/Ultralytics"
 
 from fastapi import FastAPI, File, UploadFile, HTTPException
 from contextlib import asynccontextmanager
@@ -21,6 +21,9 @@ import pandas as pd
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
+# Suppress Ultralytics logging to reduce redundant messages
+logging.getLogger("ultralytics").setLevel(logging.WARNING)
+
 # Log environment variable for debugging
 logger.info(f"YOLO_CONFIG_DIR set to: {os.getenv('YOLO_CONFIG_DIR')}")
 
@@ -57,6 +60,7 @@ app = FastAPI(lifespan=lifespan)
 class YOLOv8Model:
     def __init__(self, model_path='yolov8n.pt'):
         try:
+            logger.info("Initializing YOLOv8 model")
             self.model = YOLO(model_path)  # Load YOLOv8 model
             logger.info("YOLOv8 model loaded successfully")
         except Exception as e: