Spanicin commited on
Commit
46d2820
·
verified ·
1 Parent(s): 3d86c74

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ data_ready = False # Flag to check if extraction is complete
30
  lock = threading.Lock() # Lock to manage concurrent access
31
  extracted_texts = {}
32
 
33
- os.environ["TRANSFORMERS_CACHE"] = "/app/cache"
34
  ocr_tokenizer = AutoTokenizer.from_pretrained('ucaslcl/GOT-OCR2_0', trust_remote_code=True)
35
  ocr_model = AutoModel.from_pretrained(
36
  'ucaslcl/GOT-OCR2_0', trust_remote_code=True,
 
30
  lock = threading.Lock() # Lock to manage concurrent access
31
  extracted_texts = {}
32
 
33
+ os.environ["HF_HOME"] = "/app/cache"
34
  ocr_tokenizer = AutoTokenizer.from_pretrained('ucaslcl/GOT-OCR2_0', trust_remote_code=True)
35
  ocr_model = AutoModel.from_pretrained(
36
  'ucaslcl/GOT-OCR2_0', trust_remote_code=True,