Ermine-e commited on
Commit
b93145d
1 Parent(s): afb9b7f

Update app.py

Browse files

Changed reference to the new Chroma database with annotated transcripts. Should improve the quality of responses.
Also changed the number of chunks from 4 to 5.

Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -27,7 +27,7 @@ def load_db(k):
27
  openai_api_version="2023-05-15",
28
  )
29
  # download Chroma vector database
30
- persist_directory = 'chroma_1500ch_vada/'
31
  db = Chroma(persist_directory=persist_directory,
32
  embedding_function=embeddings)
33
  # define LLM
@@ -119,7 +119,7 @@ class cbfs(param.Parameterized):
119
  def __init__(self, **params):
120
  super(cbfs, self).__init__(**params)
121
  self.panels = []
122
- self.qa = load_db(4)
123
  self.clr_history()
124
 
125
  def convchain(self, query):
 
27
  openai_api_version="2023-05-15",
28
  )
29
  # download Chroma vector database
30
+ persist_directory = 'chroma_1500ch_new/'
31
  db = Chroma(persist_directory=persist_directory,
32
  embedding_function=embeddings)
33
  # define LLM
 
119
  def __init__(self, **params):
120
  super(cbfs, self).__init__(**params)
121
  self.panels = []
122
+ self.qa = load_db(5) # indicate number of chunks
123
  self.clr_history()
124
 
125
  def convchain(self, query):