ykl7 commited on
Commit
1df6d5c
·
1 Parent(s): 5fa5d29

improve retrieved document parsing

Browse files
Files changed (1) hide show
  1. app.py +6 -2
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
- title = document[:document.lower().index("abstract: ")].replace('\n', ' ')
150
- retrieved_document_titles = f"{retrieved_document_titles}\n"
 
 
 
 
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