Bennie12 commited on
Commit
7cd5f22
·
verified ·
1 Parent(s): c5ae22c

Update bert_explainer.py

Browse files
Files changed (1) hide show
  1. bert_explainer.py +6 -0
bert_explainer.py CHANGED
@@ -12,6 +12,12 @@ from transformers import BertTokenizer
12
  from AI_Model_architecture import BertLSTM_CNN_Classifier
13
  from lime.lime_text import LimeTextExplainer
14
 
 
 
 
 
 
 
15
  os.environ["HF_HOME"] = "/tmp/huggingface"
16
  os.environ["HF_DATASETS_CACHE"] = "/tmp/huggingface/datasets"
17
  os.environ["HF_TRANSFORMERS_CACHE"] = "/tmp/huggingface/transformers"
 
12
  from AI_Model_architecture import BertLSTM_CNN_Classifier
13
  from lime.lime_text import LimeTextExplainer
14
 
15
+ # HuggingFace 完整快取路徑設定
16
+ os.environ["HUGGINGFACE_HUB_CACHE"] = "/tmp/huggingface_cache"
17
+
18
+ # EasyOCR 路徑設定
19
+ os.environ["EASYOCR_MODULE_PATH"] = "/tmp/easyocr"
20
+
21
  os.environ["HF_HOME"] = "/tmp/huggingface"
22
  os.environ["HF_DATASETS_CACHE"] = "/tmp/huggingface/datasets"
23
  os.environ["HF_TRANSFORMERS_CACHE"] = "/tmp/huggingface/transformers"