Update app.py

app.py

@@ -2,11 +2,21 @@ import pandas as pd
 import faiss
 import numpy as np
 from sentence_transformers import SentenceTransformer
+from datasets import load_dataset
+from datasets import concatenate_datasets
 
-# Загружаем датасет с вопросами и ответами
-train_df = pd.read_csv("data/ostap_train.csv")
-questions = train_df["question"].tolist()
-answers = train_df["answer"].tolist()
+# Load dataset from Hugging Face Hub
+dataset = load_dataset("fitlemon/ostap-phrases")
+
+# Access train/test splits
+train_data = dataset["train"]
+test_data = dataset["test"]
+
+# concat datasets
+dataset = concatenate_datasets([train_data, test_data])
+
+questions = dataset["question"]
+answers = dataset["answer"]
 
 # Инициализируем эмбеддинг-модель
 model_bge = SentenceTransformer("fitlemon/bge-m3-ru-ostap")
@@ -46,4 +56,4 @@ with gr.Blocks() as app:
     msg.submit(respond, [msg, chatbot], [msg, chatbot])
 
 if __name__ == "__main__":
-    app.launch()
+    app.launch()