Milkfish033 commited on
Commit
2cffd9a
·
verified ·
1 Parent(s): 9b87900

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -1,6 +1,10 @@
1
  import os
2
- # 解决 libgomp OMP_NUM_THREADS 非法值
3
- os.environ["OMP_NUM_THREADS"] = str(int(os.getenv("OMP_NUM_THREADS", "1") or "1"))
 
 
 
 
4
 
5
  import threading
6
  import gradio as gr
 
1
  import os
2
+
3
+ # ---- Robust fix for OMP_NUM_THREADS (HF / K8s may set it to '7500m') ----
4
+ _raw_omp = os.getenv("OMP_NUM_THREADS", "")
5
+ if not _raw_omp.isdigit():
6
+ os.environ["OMP_NUM_THREADS"] = "1"
7
+
8
 
9
  import threading
10
  import gradio as gr