Npc5 commited on
Commit
9b8c45d
·
verified ·
1 Parent(s): 899d9b3

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +14 -3
app.py CHANGED
@@ -1,4 +1,18 @@
1
  import os
 
 
 
 
 
 
 
 
 
 
 
 
 
 
2
  import re
3
  import torch
4
  import spaces
@@ -30,9 +44,6 @@ def _ensure_valid_omp_threads() -> None:
30
  os.environ["OMP_NUM_THREADS"] = "1"
31
 
32
 
33
- _ensure_valid_omp_threads()
34
-
35
-
36
  def _normalize(text: str) -> str:
37
  return _RE_WS.sub(" ", (text or "").replace("\r", " ").replace("\n", " ")).strip()
38
 
 
1
  import os
2
+
3
+
4
+ def _ensure_valid_omp_threads() -> None:
5
+ # Some Spaces images set OMP_NUM_THREADS to invalid values; libgomp will error.
6
+ v = (os.environ.get("OMP_NUM_THREADS") or "").strip()
7
+ try:
8
+ if int(v) <= 0:
9
+ raise ValueError
10
+ except Exception:
11
+ os.environ["OMP_NUM_THREADS"] = "1"
12
+
13
+
14
+ _ensure_valid_omp_threads()
15
+
16
  import re
17
  import torch
18
  import spaces
 
44
  os.environ["OMP_NUM_THREADS"] = "1"
45
 
46
 
 
 
 
47
  def _normalize(text: str) -> str:
48
  return _RE_WS.sub(" ", (text or "").replace("\r", " ").replace("\n", " ")).strip()
49