Update app.py

app.py

@@ -40,7 +40,7 @@ class DocumentSearch:
         # perform search via faiss FlatIP index
         distances, indeces = self.index.search(query_vector, k*10)
         # get docs by index
-        res_docs = [self.docs[i] for i in indeces[0]]
+        res_docs = [self.docs[min(i, len(self.docs) - 1)] for i in indeces[0]]
         # get scores by index
         dists = [dist for dist in distances[0]]