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

Update bert_explainer.py

Browse files
Files changed (1) hide show
  1. bert_explainer.py +3 -2
bert_explainer.py CHANGED
@@ -11,12 +11,13 @@ from huggingface_hub import hf_hub_download
11
  from transformers import BertTokenizer
12
  from AI_Model_architecture import BertLSTM_CNN_Classifier
13
  from lime.lime_text import LimeTextExplainer
 
14
  os.environ["HF_HOME"] = "/tmp/huggingface"
 
 
15
  HF_TOKEN = os.environ.get("HF_TOKEN")
16
 
17
 
18
-
19
-
20
  # OCR 模組
21
  import os
22
 
 
11
  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"
18
  HF_TOKEN = os.environ.get("HF_TOKEN")
19
 
20
 
 
 
21
  # OCR 模組
22
  import os
23