Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -35,11 +35,9 @@ async def setup_endpoint(req: SetupRequest):
|
|
| 35 |
|
| 36 |
# Create ChromaDB collection if not exists
|
| 37 |
chroma_client = chromadb.Client(
|
| 38 |
-
|
| 39 |
-
|
| 40 |
-
|
| 41 |
-
)
|
| 42 |
-
)
|
| 43 |
|
| 44 |
try:
|
| 45 |
kb_collection = chroma_client.get_collection("kb_collection")
|
|
|
|
| 35 |
|
| 36 |
# Create ChromaDB collection if not exists
|
| 37 |
chroma_client = chromadb.Client(
|
| 38 |
+
persist_directory="/data/chroma_db", # where DB will be stored
|
| 39 |
+
chroma_db_impl="duckdb+parquet"
|
| 40 |
+
)
|
|
|
|
|
|
|
| 41 |
|
| 42 |
try:
|
| 43 |
kb_collection = chroma_client.get_collection("kb_collection")
|