No FA

app.py

@@ -3,13 +3,13 @@ MCPyLate Server
 A Model Context Protocol server that provides search functionality using PyLate.
 """
 
-import subprocess
+# import subprocess
 
-subprocess.run(
-    "pip install flash-attn --no-build-isolation",
-    env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
-    shell=True,
-)
+# subprocess.run(
+#     "pip install flash-attn --no-build-isolation",
+#     env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
+#     shell=True,
+# )
 from typing import Any, Dict, List, Optional
 
 from core import MCPyLate
@@ -60,7 +60,11 @@ def register_tools(mcp: FastMCP, pylate: MCPyLate):
 
 def main():
     """Main entry point for the server."""
-    snapshot_download(repo_id="leetcode_Reason-ModernColBERT", local_dir="indexes/")
+    snapshot_download(
+        repo_id="lightonai/leetcode_reasonmoderncolbert",
+        local_dir="indexess/",
+        repo_type="dataset",
+    )
     try:
         mcp = FastMCP("pylate-search-server")