Enhance background preprocessing to handle Kubernetes CPU limits for rembg

app_utils.py

@@ -1,3 +1,4 @@
+import os
 import rembg
 import random
 import torch
@@ -64,6 +65,11 @@ def remove_background(image: Image,
 def background_preprocess(input_image, do_remove_background):
     if input_image is None:
         return None
+    if do_remove_background:
+        omp_num_threads = os.getenv("OMP_NUM_THREADS")
+        if omp_num_threads and omp_num_threads.endswith("m"):
+            # Kubernetes CPU limits may be expressed in millicores; rembg expects an int.
+            os.environ["OMP_NUM_THREADS"] = str(max(1, int(omp_num_threads[:-1]) // 1000))
     rembg_session = rembg.new_session() if do_remove_background else None
 
     if do_remove_background: