logan-codes commited on
Commit
ea1e7dc
·
1 Parent(s): 4e28509

changed the data dir to hf compatible

Browse files
app/services/document_ingester.py CHANGED
@@ -14,7 +14,8 @@ class Ingester:
14
  model_name="sentence-transformers/all-MiniLM-L6-v2"
15
  )
16
  load_dotenv()
17
- self.DATA_DIR = os.getenv("DATA_DIR")
 
18
  self.vector_store= Chroma(
19
  collection_name="documents_collection",
20
  embedding_function=self.embedding_model,
 
14
  model_name="sentence-transformers/all-MiniLM-L6-v2"
15
  )
16
  load_dotenv()
17
+ base_dir = os.getenv("HF_HOME", "/home/user/app")
18
+ self.DATA_DIR = os.path.join(base_dir, "data")
19
  self.vector_store= Chroma(
20
  collection_name="documents_collection",
21
  embedding_function=self.embedding_model,
app/services/retriever.py CHANGED
@@ -12,7 +12,8 @@ class Retriever:
12
  model_name="sentence-transformers/all-MiniLM-L6-v2"
13
  )
14
  load_dotenv()
15
- self.DATA_DIR = os.getenv("DATA_DIR")
 
16
  self.vector_store=Chroma(
17
  collection_name="documents_collection",
18
  embedding_function=self.embed,
 
12
  model_name="sentence-transformers/all-MiniLM-L6-v2"
13
  )
14
  load_dotenv()
15
+ base_dir = os.getenv("HF_HOME", "/home/user/app")
16
+ self.DATA_DIR = os.path.join(base_dir, "data")
17
  self.vector_store=Chroma(
18
  collection_name="documents_collection",
19
  embedding_function=self.embed,