pre_start.py

#!/usr/bin/env python3
"""Pre-start script to fix OMP_NUM_THREADS before any imports"""
import os
import subprocess
import sys

# Fix OMP_NUM_THREADS FIRST, before any other imports
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'")

# Now run streamlit with proper HF Spaces config
sys.exit(subprocess.call([
    'streamlit', 'run', 'app.py',
    '--server.port=7860',
    '--server.address=0.0.0.0',
    '--server.headless=true',
    '--browser.serverAddress=0.0.0.0',
    '--browser.gatherUsageStats=false'
]))