Update app.py

app.py

@@ -6,6 +6,14 @@ from langchain.vectorstores import FAISS
 from langchain_huggingface import HuggingFaceEmbeddings
 from tqdm import tqdm
 
+
+# load env
+from dotenv import load_dotenv
+
+load_dotenv()
+
+key = os.getenv("HF_KEY")
+
 # Путь для сохранения FAISS-индекса
 INDEX_PATH = "./faiss_index"
 
@@ -19,8 +27,8 @@ def update_faiss_index():
     создаёт FAISS-индекс и сохраняет его локально.
     """
     # Загружаем датасет (например, сплит "train")
-    train_dataset = load_dataset("fitlemon/rag-labor-codex-dataset")["train"]
-    test_dataset = load_dataset("fitlemon/rag-labor-codex-dataset")["test"]
+    train_dataset = load_dataset("fitlemon/rag-labor-codex-dataset", token=key)["train"]
+    test_dataset = load_dataset("fitlemon/rag-labor-codex-dataset", token=key)["test"]
     # combine train and test datasets
     dataset = concatenate_datasets([train_dataset, test_dataset])
     # get rid off duplicate chunks
@@ -93,4 +101,4 @@ iface = gr.Interface(
 )
 
 if __name__ == "__main__":
-    iface.launch()
+    iface.launch()