Spaces:
Sleeping
Sleeping
Commit
·
c27dcb7
1
Parent(s):
1136c97
Fix streamlit config for HF Spaces routing
Browse files- pre_start.py +11 -3
pre_start.py
CHANGED
|
@@ -1,6 +1,7 @@
|
|
| 1 |
#!/usr/bin/env python3
|
| 2 |
"""Pre-start script to fix OMP_NUM_THREADS before any imports"""
|
| 3 |
import os
|
|
|
|
| 4 |
import sys
|
| 5 |
|
| 6 |
# Fix OMP_NUM_THREADS FIRST, before any other imports
|
|
@@ -8,7 +9,14 @@ if 'OMP_NUM_THREADS' in os.environ:
|
|
| 8 |
value = str(os.environ['OMP_NUM_THREADS'])
|
| 9 |
if not value.isdigit():
|
| 10 |
os.environ['OMP_NUM_THREADS'] = '4'
|
| 11 |
-
print(f"
|
| 12 |
|
| 13 |
-
# Now run streamlit
|
| 14 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
#!/usr/bin/env python3
|
| 2 |
"""Pre-start script to fix OMP_NUM_THREADS before any imports"""
|
| 3 |
import os
|
| 4 |
+
import subprocess
|
| 5 |
import sys
|
| 6 |
|
| 7 |
# Fix OMP_NUM_THREADS FIRST, before any other imports
|
|
|
|
| 9 |
value = str(os.environ['OMP_NUM_THREADS'])
|
| 10 |
if not value.isdigit():
|
| 11 |
os.environ['OMP_NUM_THREADS'] = '4'
|
| 12 |
+
print(f"Fixed OMP_NUM_THREADS from '{value}' to '4'")
|
| 13 |
|
| 14 |
+
# Now run streamlit with proper HF Spaces config
|
| 15 |
+
sys.exit(subprocess.call([
|
| 16 |
+
'streamlit', 'run', 'app.py',
|
| 17 |
+
'--server.port=7860',
|
| 18 |
+
'--server.address=0.0.0.0',
|
| 19 |
+
'--server.headless=true',
|
| 20 |
+
'--browser.serverAddress=0.0.0.0',
|
| 21 |
+
'--browser.gatherUsageStats=false'
|
| 22 |
+
]))
|