0504ankitsharma commited on
Commit
c4e6640
·
verified ·
1 Parent(s): 437ff30

Update app/main.py

Browse files
Files changed (1) hide show
  1. app/main.py +1 -1
app/main.py CHANGED
@@ -15,7 +15,7 @@ from langchain_community.document_loaders import UnstructuredWordDocumentLoader
15
  from langchain_community.embeddings import HuggingFaceBgeEmbeddings
16
 
17
  # Configure NLTK custom download directory
18
- NLTK_DATA_PATH = os.path.join(os.path.expanduser("~"), "nltk_data")
19
  os.makedirs(NLTK_DATA_PATH, exist_ok=True)
20
  nltk.data.path.append(NLTK_DATA_PATH)
21
 
 
15
  from langchain_community.embeddings import HuggingFaceBgeEmbeddings
16
 
17
  # Configure NLTK custom download directory
18
+ NLTK_DATA_PATH = os.getenv("NLTK_DATA_PATH", os.path.join(os.getcwd(), "nltk_data"))
19
  os.makedirs(NLTK_DATA_PATH, exist_ok=True)
20
  nltk.data.path.append(NLTK_DATA_PATH)
21