Use Python wrapper to fix OMP_NUM_THREADS environment issue

README.md

@@ -5,7 +5,7 @@ colorFrom: green
 colorTo: blue
 sdk: streamlit
 sdk_version: "1.23.1"
-app_file: startup.sh
+app_file: streamlit_wrapper.py
 pinned: false
 ---
 

streamlit_wrapper.py

@@ -0,0 +1,19 @@
+#!/usr/bin/env python3
+"""
+Wrapper script to fix HuggingFace Spaces environment issues before starting Streamlit
+"""
+import os
+import sys
+
+# Fix OMP_NUM_THREADS before ANY imports that might use it
+if 'OMP_NUM_THREADS' in os.environ:
+    omp_value = os.environ['OMP_NUM_THREADS']
+    if 'm' in str(omp_value).lower() or not str(omp_value).isdigit():
+        os.environ['OMP_NUM_THREADS'] = '4'
+        print(f"Fixed OMP_NUM_THREADS from '{omp_value}' to '4'")
+
+# Now start Streamlit
+if __name__ == '__main__':
+    from streamlit.web import cli as stcli
+    sys.argv = ["streamlit", "run", "app.py"]
+    sys.exit(stcli.main())