fix early startup thread and ORT warnings

Normalize OpenMP-related env vars before importing torch and set ONNX Runtime logger severity before rembg initializes so the Space starts with fewer container-specific warnings.

Made-with: Cursor

app.py

@@ -1,8 +1,32 @@
+import os
+
+
+def _normalize_thread_env_var(name):
+    raw_value = os.getenv(name)
+    if raw_value is None:
+        return
+
+    value = raw_value.strip()
+    if value.endswith("m"):
+        try:
+            os.environ[name] = str(max(1, int(value[:-1]) // 1000))
+        except ValueError:
+            os.environ.pop(name, None)
+        return
+
+    try:
+        os.environ[name] = str(max(1, int(value)))
+    except ValueError:
+        os.environ.pop(name, None)
+
+
+for env_name in ("OMP_NUM_THREADS", "OPENBLAS_NUM_THREADS", "MKL_NUM_THREADS", "NUMEXPR_NUM_THREADS"):
+    _normalize_thread_env_var(env_name)
+
 import gradio as gr
 import numpy as np
 from PIL import Image
 import torch
-import os
 import tempfile
 import io
 from fastapi import FastAPI, File, Form, HTTPException, UploadFile

app_utils.py

@@ -23,6 +23,14 @@ def _normalize_thread_env_var(name):
 for env_name in ("OMP_NUM_THREADS", "OPENBLAS_NUM_THREADS", "MKL_NUM_THREADS", "NUMEXPR_NUM_THREADS"):
     _normalize_thread_env_var(env_name)
 
+try:
+    import onnxruntime as ort
+
+    if hasattr(ort, "set_default_logger_severity"):
+        ort.set_default_logger_severity(3)
+except Exception:
+    ort = None
+
 import rembg
 import random
 import torch