fix: faiss

app.py

@@ -16,8 +16,8 @@ def load_models():
     retriever = RagRetriever.from_pretrained(
         "facebook/rag-sequence-nq",
         index_name="custom",
-        passages_path="/data/rag_dataset/dataset",
-        index_path="/data/rag_dataset/embeddings.faiss"
+        passages_path="rag_dataset/dataset",
+        index_path="rag_dataset/embeddings.faiss"
     )
     model = RagSequenceForGeneration.from_pretrained(
         "facebook/rag-sequence-nq",
@@ -28,7 +28,7 @@ def load_models():
 
 @st.cache_data  # Cache dataset on disk
 def load_dataset():
-    return load_from_disk("/data/rag_dataset/dataset")
+    return load_from_disk("rag_dataset/dataset")
 
 # RAG Pipeline
 def rag_pipeline(query, dataset, index):

faiss_index/index.py

@@ -23,7 +23,7 @@ def fetch_arxiv_papers(query, max_results=10):
     logging.info(f"Fetched {len(papers)} papers from arXiv")
     return papers
 
-def build_faiss_index(papers, dataset_dir="/data/rag_dataset"):
+def build_faiss_index(papers, dataset_dir="rag_dataset"):
     """Build and save dataset with FAISS index for RAG"""
     # Initialize DPR encoder
     ctx_encoder = DPRContextEncoder.from_pretrained("facebook/dpr-ctx_encoder-single-nq-base")