Upload app.py
Browse files
app.py
CHANGED
|
@@ -33,6 +33,7 @@ client = OpenAI(
|
|
| 33 |
# Define the embedding model and the vectorstore
|
| 34 |
embedding_model = SentenceTransformerEmbeddings(model_name='thenlper/gte-large')
|
| 35 |
# Load the persisted vectorDB
|
|
|
|
| 36 |
reportdb = Chroma(
|
| 37 |
collection_name=collection_name,
|
| 38 |
persist_directory='./report_db1',
|
|
|
|
| 33 |
# Define the embedding model and the vectorstore
|
| 34 |
embedding_model = SentenceTransformerEmbeddings(model_name='thenlper/gte-large')
|
| 35 |
# Load the persisted vectorDB
|
| 36 |
+
collection_name = 'Dataset-10k'
|
| 37 |
reportdb = Chroma(
|
| 38 |
collection_name=collection_name,
|
| 39 |
persist_directory='./report_db1',
|