cnmoro commited on
Commit
90f8e12
1 Parent(s): 29395ca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -110,7 +110,7 @@ def index_and_search(query, pdf_metadata):
110
  retrieved_contents[title] = []
111
  retrieved_contents[title].append(ret_cont['sentence'])
112
 
113
- retrieved_contents = {k: '\n'.join(v) for k, v in retrieved_contents.items()}
114
 
115
  return retrieved_contents, embedding_time, retrieval_time
116
 
 
110
  retrieved_contents[title] = []
111
  retrieved_contents[title].append(ret_cont['sentence'])
112
 
113
+ retrieved_contents = {k: '\n'.join(v) for k, v in retrieved_contents.items() if len(v) > 2}
114
 
115
  return retrieved_contents, embedding_time, retrieval_time
116