Anothervin1 commited on
Commit
02c6956
·
verified ·
1 Parent(s): 16e56db

Update src/embedding.py

Browse files
Files changed (1) hide show
  1. src/embedding.py +1 -1
src/embedding.py CHANGED
@@ -4,7 +4,7 @@ from sentence_transformers import SentenceTransformer
4
  from config.settings import Settings
5
 
6
  # Chỉ định thư mục cache
7
- os.environ["TRANSFORMERS_CACHE"] = "/tmp/cache"
8
 
9
  class Embedding:
10
  def __init__(self):
 
4
  from config.settings import Settings
5
 
6
  # Chỉ định thư mục cache
7
+ os.environ["HF_HOME"] = "/tmp/cache"
8
 
9
  class Embedding:
10
  def __init__(self):