tech5 commited on
Commit
66cafc7
·
1 Parent(s): 6048b8c

Copied GitHub project to Hugging Face Space

Browse files
Files changed (1) hide show
  1. backend/models/embed.py +5 -0
backend/models/embed.py CHANGED
@@ -1,3 +1,8 @@
 
 
 
 
 
1
  from langchain_community.document_loaders import PyPDFDirectoryLoader
2
  from langchain.text_splitter import RecursiveCharacterTextSplitter
3
  from langchain_community.embeddings import HuggingFaceEmbeddings
 
1
+ import os
2
+ os.environ['TRANSFORMERS_CACHE'] = '/tmp/hf_cache'
3
+ os.environ['HF_HOME'] = '/tmp/hf_home'
4
+ os.environ['SENTENCE_TRANSFORMERS_HOME'] = '/tmp/st_cache'
5
+
6
  from langchain_community.document_loaders import PyPDFDirectoryLoader
7
  from langchain.text_splitter import RecursiveCharacterTextSplitter
8
  from langchain_community.embeddings import HuggingFaceEmbeddings