Update app.py

app.py

@@ -20,17 +20,20 @@ import pandas as pd
 # Suppress all warnings
 warnings.filterwarnings("ignore")
 
-# Set up YOLO_CONFIG_DIR
+# Set up YOLO_CONFIG_DIR only once
 yolo_config_dir = "/tmp/Ultralytics"
-try:
-    if os.path.exists(yolo_config_dir):
-        shutil.rmtree(yolo_config_dir)
-    os.makedirs(yolo_config_dir, exist_ok=True)
-    os.chmod(yolo_config_dir, 0o777)  # Ensure directory is writable
-    os.environ["YOLO_CONFIG_DIR"] = yolo_config_dir
-except Exception as e:
-    print(f"Failed to set up YOLO_CONFIG_DIR: {e}")
-    raise
+if not hasattr(sys, '_yolo_config_initialized'):
+    try:
+        if os.path.exists(yolo_config_dir):
+            shutil.rmtree(yolo_config_dir)
+        os.makedirs(yolo_config_dir, exist_ok=True)
+        os.chmod(yolo_config_dir, 0o777)  # Ensure directory is writable
+        os.environ["YOLO_CONFIG_DIR"] = yolo_config_dir
+        sys._yolo_config_initialized = True
+        print(f"YOLO_CONFIG_DIR initialized to: {yolo_config_dir}")
+    except Exception as e:
+        print(f"Failed to set up YOLO_CONFIG_DIR: {e}")
+        raise
 
 # --- Logging Configuration ---
 logging.basicConfig(
@@ -48,7 +51,7 @@ logging.getLogger("PIL").setLevel(logging.WARNING)
 logging.getLogger("matplotlib").setLevel(logging.WARNING)
 logging.getLogger("urllib3").setLevel(logging.WARNING)
 
-# Log environment variable once
+# Log environment variable
 logger.info(f"YOLO_CONFIG_DIR set to: {os.getenv('YOLO_CONFIG_DIR')}")
 
 # --- Environment Variables ---
@@ -63,8 +66,12 @@ yolo_model = None
 
 # --- Model Class ---
 class YOLOv8Model:
-    def __init__(self, model_path='yolov8n.pt'):
+    def __init__(self, model_path='/app/yolov8n.pt'):
         try:
+            logger.info("Checking for model weights at %s", model_path)
+            if not os.path.exists(model_path):
+                logger.error("Model weights file %s not found", model_path)
+                raise FileNotFoundError(f"Model weights file {model_path} not found")
             logger.info("Initializing YOLOv8 model")
             original_stdout = sys.stdout
             sys.stdout = open(os.devnull, 'w')