shevadesuyash commited on
Commit
e8cccdd
·
verified ·
1 Parent(s): 0cf2485

Update paragraph_checker.py

Browse files
Files changed (1) hide show
  1. paragraph_checker.py +6 -2
paragraph_checker.py CHANGED
@@ -1,6 +1,10 @@
1
- import os
 
2
  os.environ['HOME'] = '/app'
3
- os.environ['LANGTOOL_HOME'] = '/app/.cache/languagetool'
 
 
 
4
 
5
  import language_tool_python
6
  from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM
 
1
+
2
+ # Set cache dirs (must match Dockerfile env vars)
3
  os.environ['HOME'] = '/app'
4
+ os.environ['HF_HOME'] = '/app/.hf_cache'
5
+ os.environ['LANGTOOL_HOME'] = '/app/.ltool_cache'
6
+ os.environ['XDG_CACHE_HOME'] = '/app/.cache'
7
+
8
 
9
  import language_tool_python
10
  from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM