updated server

app.py

@@ -9,6 +9,7 @@ import pandas as pd
 from langchain.docstore.document import Document
 from langchain.text_splitter import RecursiveCharacterTextSplitter
 from langchain_community.retrievers import BM25Retriever
+from datasets import load_dataset
 
 # Load environment variables
 load_dotenv()
@@ -33,7 +34,8 @@ class GitHubMCPServer:
     def _initialize_cve_retriever(self):
         """Initialize the CVE retriever with the dataset"""
         try:
-            csv_path = "dataset/cve.csv"
+            csv_path = load_dataset("Baction/cve")
+            
             if not os.path.exists(csv_path):
                 print(f"⚠ CVE dataset not found at {csv_path}")
                 return