Update app.py

app.py

@@ -9,7 +9,7 @@ import torch
 
 index = faiss.read_index("index.faiss")
 
-with open("texts.pkl", "rb") as f:
+with open("texts_long.pkl", "rb") as f:
     texts = pickle.load(f)
 
 
@@ -55,7 +55,7 @@ def retrieve(query, k=3):
 
     scores, ids = index.search(q, k)
 
-    return [texts[i] for i in ids[0]]
+    return [str(i) +": " + texts[i] + "\n" for i in ids[0]]
 
 
 with gr.Blocks() as demo: