button added

app.py

@@ -132,8 +132,8 @@ if prompt := st.chat_input(initial_input):
         ref = pd.DataFrame(
             {
                 "idx": idx,
-                "questions": [dataset["train"]["questions"][i] for i in idx],
-                "answers": [dataset["train"]["answers"][i] for i in idx],
+                "questions": [dataset["train"]["questions"][0:20][i] for i in idx],
+                "answers": [dataset["train"]["answers"][0:20][i] for i in idx],
                 "distances": results["distances"][0],
             }
         )