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