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 |
-
|
|
|
|
| 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 |
-
|
|
|
|
| 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,
|