Spaces:
Sleeping
Sleeping
Commit
·
0ec0bdb
1
Parent(s):
6968e13
chromaDB
Browse files- Dockerfile +1 -1
- data_ingetion/vectroDB.py +6 -0
- requirements.txt +1 -0
Dockerfile
CHANGED
|
@@ -19,4 +19,4 @@ RUN mkdir -p /app/data && \
|
|
| 19 |
|
| 20 |
COPY . /app
|
| 21 |
|
| 22 |
-
CMD ["uvicorn", "
|
|
|
|
| 19 |
|
| 20 |
COPY . /app
|
| 21 |
|
| 22 |
+
CMD ["uvicorn", "main_api:app", "--host", "0.0.0.0", "--port", "7860"]
|
data_ingetion/vectroDB.py
CHANGED
|
@@ -5,6 +5,12 @@ import pypdf
|
|
| 5 |
import uuid
|
| 6 |
import os
|
| 7 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 8 |
VECTOR_NAME = "database"
|
| 9 |
EMBEDDING_MODEL = "togethercomputer/m2-bert-80M-2k-retrieval"
|
| 10 |
|
|
|
|
| 5 |
import uuid
|
| 6 |
import os
|
| 7 |
|
| 8 |
+
# for chromaDB
|
| 9 |
+
__import__("pysqlite3")
|
| 10 |
+
import sys
|
| 11 |
+
|
| 12 |
+
sys.modules["sqlite3"] = sys.modules.pop("pysqlite3")
|
| 13 |
+
|
| 14 |
VECTOR_NAME = "database"
|
| 15 |
EMBEDDING_MODEL = "togethercomputer/m2-bert-80M-2k-retrieval"
|
| 16 |
|
requirements.txt
CHANGED
|
@@ -109,6 +109,7 @@ Pygments==2.19.1
|
|
| 109 |
pypdf==5.5.0
|
| 110 |
PyPika==0.48.9
|
| 111 |
pyproject_hooks==1.2.0
|
|
|
|
| 112 |
python-dateutil==2.9.0.post0
|
| 113 |
python-dotenv==1.1.0
|
| 114 |
python-multipart==0.0.20
|
|
|
|
| 109 |
pypdf==5.5.0
|
| 110 |
PyPika==0.48.9
|
| 111 |
pyproject_hooks==1.2.0
|
| 112 |
+
pysqlite3-binary==0.5.3
|
| 113 |
python-dateutil==2.9.0.post0
|
| 114 |
python-dotenv==1.1.0
|
| 115 |
python-multipart==0.0.20
|