Update app.py

app.py

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