ravi6k commited on
Commit
9598589
·
verified ·
1 Parent(s): 1dd91c4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -28,12 +28,13 @@ client = OpenAI(
28
  embedding_model_name = 'thenlper/gte-large'
29
  embedding_model = SentenceTransformerEmbeddings(model_name=embedding_model_name)
30
  collection_name_qna = 'report_10K_db'
31
- persisted_vectordb_location = './report_10K_db/report_10K_db'
 
32
  # Load the persisted vectorDB
33
  vectorstore_persisted = Chroma(
34
  collection_name=collection_name_qna,
35
  persist_directory=persisted_vectordb_location,
36
- embedding_function=embedding_model
37
  )
38
 
39
  # Prepare the logging functionality
 
28
  embedding_model_name = 'thenlper/gte-large'
29
  embedding_model = SentenceTransformerEmbeddings(model_name=embedding_model_name)
30
  collection_name_qna = 'report_10K_db'
31
+ persisted_vectordb_location = './report_10K_db'
32
+
33
  # Load the persisted vectorDB
34
  vectorstore_persisted = Chroma(
35
  collection_name=collection_name_qna,
36
  persist_directory=persisted_vectordb_location,
37
+ embedding_function='thenlper/gte-large'
38
  )
39
 
40
  # Prepare the logging functionality