danidanidani commited on
Commit
c27dcb7
·
1 Parent(s): 1136c97

Fix streamlit config for HF Spaces routing

Browse files
Files changed (1) hide show
  1. 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"Fixed OMP_NUM_THREADS from '{value}' to '4'")
12
 
13
- # Now run streamlit
14
- os.execlp('streamlit', 'streamlit', 'run', 'app.py')
 
 
 
 
 
 
 
 
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
+ ]))