devve1 commited on
Commit
7986ab0
1 Parent(s): fb9d31c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -6
app.py CHANGED
@@ -134,8 +134,6 @@ def main(query: str, client: QdrantClient, collection_name: str, llm, dense_mode
134
  dense_query = list(dense_model.embed_query(query, 32))
135
  sparse_query = list(sparse_model.embed(query, 32))
136
 
137
- print('THHFGHFGBFGBFB')
138
-
139
  search_results = search(
140
  client,
141
  collection_name,
@@ -162,8 +160,6 @@ def main(query: str, client: QdrantClient, collection_name: str, llm, dense_mode
162
  input_variables=['text']
163
  )
164
 
165
- print('GHNTBECDFVFDGFDGDG')
166
-
167
  map_chain = load_summarize_chain(llm,
168
  "stuff",
169
  prompt=map_prompt
@@ -171,13 +167,12 @@ def main(query: str, client: QdrantClient, collection_name: str, llm, dense_mode
171
 
172
  summary_list = []
173
 
174
- print('UYKUIOOOILOLI')
175
-
176
  for doc in docs:
177
  chunk_summary = map_chain.invoke([doc])
178
  summary_list.append(chunk_summary['output_text'])
179
 
180
  summaries = Document(page_content="\n".join(summary_list))
 
181
 
182
  reduce_chain = load_summarize_chain(llm,
183
  "stuff",
@@ -185,6 +180,7 @@ def main(query: str, client: QdrantClient, collection_name: str, llm, dense_mode
185
  )
186
 
187
  output = reduce_chain.invoke([summaries])
 
188
  return output['output_text']
189
 
190
  @st.cache_resource
 
134
  dense_query = list(dense_model.embed_query(query, 32))
135
  sparse_query = list(sparse_model.embed(query, 32))
136
 
 
 
137
  search_results = search(
138
  client,
139
  collection_name,
 
160
  input_variables=['text']
161
  )
162
 
 
 
163
  map_chain = load_summarize_chain(llm,
164
  "stuff",
165
  prompt=map_prompt
 
167
 
168
  summary_list = []
169
 
 
 
170
  for doc in docs:
171
  chunk_summary = map_chain.invoke([doc])
172
  summary_list.append(chunk_summary['output_text'])
173
 
174
  summaries = Document(page_content="\n".join(summary_list))
175
+ print('THHFGHFGBFGBFB')
176
 
177
  reduce_chain = load_summarize_chain(llm,
178
  "stuff",
 
180
  )
181
 
182
  output = reduce_chain.invoke([summaries])
183
+ print('ADSDGFGVHBTFEFSDGFES')
184
  return output['output_text']
185
 
186
  @st.cache_resource