akryldigital commited on
Commit
039f258
·
verified ·
1 Parent(s): 008269a

fix ENV vars

Browse files
Files changed (1) hide show
  1. app.py +28 -0
app.py CHANGED
@@ -23,6 +23,34 @@ from src.reporting.feedback_schema import create_feedback_from_dict
23
  logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
24
  logger = logging.getLogger(__name__)
25
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
26
  # Page config
27
  st.set_page_config(
28
  layout="wide",
 
23
  logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
24
  logger = logging.getLogger(__name__)
25
 
26
+
27
+
28
+ cache_dir = Path("/app/.cache/huggingface")
29
+ cache_dir.mkdir(parents=True, exist_ok=True)
30
+ os.environ["HF_HOME"] = str(cache_dir)
31
+ os.environ["TRANSFORMERS_CACHE"] = str(cache_dir)
32
+ os.environ["HF_DATASETS_CACHE"] = str(cache_dir)
33
+ os.environ["HF_HUB_CACHE"] = str(cache_dir)
34
+ os.environ["SENTENCE_TRANSFORMERS_HOME"] = str(cache_dir)
35
+
36
+
37
+ omp_threads = os.environ.get("OMP_NUM_THREADS", "")
38
+ try:
39
+ # If set, validate it's a positive integer
40
+ if omp_threads:
41
+ threads = int(omp_threads)
42
+ if threads <= 0:
43
+ os.environ["OMP_NUM_THREADS"] = "1"
44
+ logger.warning(f"OMP_NUM_THREADS={omp_threads} is invalid, setting to 1")
45
+ else:
46
+ # If not set, set a reasonable default
47
+ os.environ["OMP_NUM_THREADS"] = "1"
48
+ except (ValueError, TypeError):
49
+ # If invalid format, set to default
50
+ os.environ["OMP_NUM_THREADS"] = "1"
51
+ logger.warning(f"OMP_NUM_THREADS={omp_threads} is invalid, setting to 1")
52
+
53
+
54
  # Page config
55
  st.set_page_config(
56
  layout="wide",