Spaces:
Running
Running
improve retrieved document parsing
Browse files
app.py
CHANGED
@@ -146,8 +146,12 @@ def retriever(query: str, selected_retriever: str):
|
|
146 |
|
147 |
retrieved_document_titles = ""
|
148 |
for document in topk_documents:
|
149 |
-
|
150 |
-
|
|
|
|
|
|
|
|
|
151 |
|
152 |
display_retrieved_documents = f"\nRetrieved Documents: \n{retrieved_document_titles}"
|
153 |
|
|
|
146 |
|
147 |
retrieved_document_titles = ""
|
148 |
for document in topk_documents:
|
149 |
+
try:
|
150 |
+
idx = document.lower().index("abstract: ")
|
151 |
+
title = document[:idx].replace('\n', ' ')
|
152 |
+
retrieved_document_titles = retrieved_document_titles + title + '\n'
|
153 |
+
except:
|
154 |
+
retrieved_document_titles = retrieved_document_titles + document + '\n'
|
155 |
|
156 |
display_retrieved_documents = f"\nRetrieved Documents: \n{retrieved_document_titles}"
|
157 |
|