Update app.py
Browse files
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
|