Update app.py

app.py

@@ -32,10 +32,10 @@ def update_faiss_index():
     """
     train_dataset = load_dataset("fitlemon/rag-labor-codex-dataset", token=hf_key)[
         "train"
-    ].select(4)
+    ][:2]
     test_dataset = load_dataset("fitlemon/rag-labor-codex-dataset", token=hf_key)[
         "test"
-    ].select(4)
+    ][:2]
     
     dataset = concatenate_datasets([train_dataset, test_dataset])