minor updates

app.py

@@ -44,7 +44,7 @@ class DocumentSearch:
         # get scores by index
         dists = [dist for dist in distances[0]]
         
-        return[{'doc': doc[0], 'url': doc[1], 'score': dist} for doc, dist in zip(res_docs, dists)]
+        return[{'doc': doc[0], 'url': doc[1], 'score': dist} for doc, dist in zip(res_docs, dists)][:k]
         ##### OLD VERSION WITH CROSS-ENCODER #####
         # get answers by index
         #answers = [self.docs[i] for i in indeces[0]]
@@ -85,7 +85,7 @@ if __name__ == "__main__":
         # set start time
         stt = time.time()
         # retrieve top 5 documents
-        results = surfer.search(query, k=5)
+        results = surfer.search(query, k=10)
         # set endtime
         ent = time.time()
         # measure resulting time
@@ -98,7 +98,7 @@ if __name__ == "__main__":
             # answer starts with header
             st.subheader(f"Answer {i+1}")
             # cropped answer
-            doc = answer["doc"][:150] + "..."
+            doc = answer["doc"][:250] + "..."
             # and url to the full answer
             url = answer["url"]
             # then we display it