remove req

app.py

@@ -13,6 +13,7 @@ subprocess.run(
 from typing import Any, Dict, List, Optional
 
 from core import MCPyLate
+from huggingface_hub import snapshot_download
 from mcp.server.fastmcp import FastMCP
 
 
@@ -20,10 +21,10 @@ def register_tools(mcp: FastMCP, pylate: MCPyLate):
     """Register all tools with the MCP server."""
 
     @mcp.tool(
-        name="pylate_search_msmarco",
-        description="Perform a multi-vector search on the msmarco index. Returns top‑k hits with docid, score, and snippet.",
+        name="pylate_search_leetcode",
+        description="Perform a multi-vector search on the leetcode index. Returns top‑k hits with docid, score, and snippet.",
     )
-    def pylate_search_msmarco(
+    def pylate_search_leetcode(
         query: str, k: int = 10, index_name: Optional[str] = None
     ) -> List[Dict[str, Any]]:
         """
@@ -59,6 +60,7 @@ 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/")
     try:
         mcp = FastMCP("pylate-search-server")
 

requirements.txt

@@ -1,3 +1,2 @@
 gradio
-git+https://github.com/lightonai/pylate/.git@MCPyLate
-lightonai/pylate/tree/MCPyLate
+git+https://github.com/lightonai/pylate/.git@MCPyLate