Update app.py

app.py

@@ -32,13 +32,13 @@ def update_faiss_index():
     """
     train_dataset = load_dataset("fitlemon/rag-labor-codex-dataset", token=hf_key)[
         "train"
-    ][:2]
+    ]
     test_dataset = load_dataset("fitlemon/rag-labor-codex-dataset", token=hf_key)[
         "test"
-    ][:2]
+    ]
     
     dataset = concatenate_datasets([train_dataset, test_dataset])
-
+    dataset = dataset[:3]
     docs = []
     unique_chunks = set()
     for row in tqdm(dataset, desc="Загрузка документов..."):