Update app.py

app.py

@@ -1,3 +1,8 @@
+import os
+
+# Set YOLO_CONFIG_DIR before any Ultralytics imports
+os.environ["YOLO_CONFIG_DIR"] = "/tmp"
+
 from fastapi import FastAPI, File, UploadFile, HTTPException
 from contextlib import asynccontextmanager
 import torch
@@ -6,7 +11,6 @@ import cv2
 import requests
 import json
 import time
-import os
 import numpy as np
 from pathlib import Path
 from datetime import datetime
@@ -17,8 +21,8 @@ import pandas as pd
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-# Set YOLO_CONFIG_DIR explicitly before loading Ultralytics
-os.environ["YOLO_CONFIG_DIR"] = "/tmp"
+# Log environment variable for debugging
+logger.info(f"YOLO_CONFIG_DIR set to: {os.getenv('YOLO_CONFIG_DIR')}")
 
 RTSP_URL = os.getenv("RTSP_URL", "rtsp://localhost:8554/stream")
 SALESFORCE_URL = os.getenv("SALESFORCE_URL", "https://your_salesforce_instance_url")