Add sitecustomize.py to fix OMP_NUM_THREADS at Python startup

post_install.py

@@ -0,0 +1,39 @@
+#!/usr/bin/env python3
+"""
+Post-install script to patch numexpr to handle invalid OMP_NUM_THREADS values
+"""
+import os
+import site
+
+def patch_numexpr():
+    """Find and patch numexpr/utils.py to handle invalid OMP_NUM_THREADS"""
+    for site_dir in site.getsitepackages():
+        numexpr_utils = os.path.join(site_dir, 'numexpr', 'utils.py')
+        if os.path.exists(numexpr_utils):
+            print(f"Found numexpr at: {numexpr_utils}")
+            
+            with open(numexpr_utils, 'r') as f:
+                content = f.read()
+            
+            # Replace the problematic line
+            old_code = "requested_threads = int(os.environ['OMP_NUM_THREADS'])"
+            new_code = """try:
+                requested_threads = int(os.environ['OMP_NUM_THREADS'])
+            except (ValueError, TypeError):
+                requested_threads = 4  # Default for HF Spaces T4 GPU"""
+            
+            if old_code in content and new_code not in content:
+                content = content.replace(old_code, new_code)
+                
+                with open(numexpr_utils, 'w') as f:
+                    f.write(content)
+                
+                print("✅ Patched numexpr successfully!")
+                return True
+            else:
+                print("Already patched or couldn't find code to patch")
+    
+    return False
+
+if __name__ == '__main__':
+    patch_numexpr()

requirements.txt

@@ -93,3 +93,4 @@ validators==0.20.0
 yarl==1.9.2
 zipp==3.15.0
 torch>=2.0.0
+-e .

setup.py

@@ -0,0 +1,24 @@
+from setuptools import setup
+from setuptools.command.install import install
+import os
+import shutil
+import site
+
+class PostInstallCommand(install):
+    def run(self):
+        install.run(self)
+        # Copy sitecustomize.py to site-packages
+        for site_dir in site.getsitepackages():
+            try:
+                dest = os.path.join(site_dir, 'sitecustomize.py')
+                shutil.copy('sitecustomize.py', dest)
+                print(f"✅ Installed sitecustomize.py to {dest}")
+                break
+            except:
+                continue
+
+setup(
+    name='grdn-hf-fix',
+    version='0.0.1',
+    cmdclass={'install': PostInstallCommand},
+)

sitecustomize.py

@@ -0,0 +1,12 @@
+"""
+This file runs automatically when Python starts, before any imports.
+Used to fix HuggingFace Spaces OMP_NUM_THREADS issue.
+"""
+import os
+
+# Fix OMP_NUM_THREADS before numexpr tries to use it
+if 'OMP_NUM_THREADS' in os.environ:
+    value = str(os.environ['OMP_NUM_THREADS'])
+    if not value.isdigit():
+        os.environ['OMP_NUM_THREADS'] = '4'
+        print(f"Fixed OMP_NUM_THREADS from '{value}' to '4'")