Update README.md

README.md

@@ -6,7 +6,7 @@ AgentIR-4B is a retriever specialized for Deep Research agents. Unlike conventio
 
 When employed for end-to-end Deep Research on BrowseComp-Plus, AgentIR brings substantial effectiveness and efficiency gains for agents, improving agent accuracy while reducing the number of problem-solving iterations.
 - Paper: https://arxiv.org/abs/2603.04384
-- Code: https://github.com/texttron/AgentIR
+- Code: https://github.com/texttron/AgentIR/tree/main
 - Data: https://huggingface.co/datasets/Tevatron/AgentIR-data
 - Project Page: https://texttron.github.io/AgentIR/
 
@@ -52,7 +52,7 @@ for doc, vec in zip(DOCS, docs):
     print(f"{torch.dot(q, vec).item():.6f}  {doc}")
 ```
 
-To run end-to-end Deep Research with AgentIR-4B, please see https://github.com/texttron/AgentIR.
+To run end-to-end Deep Research with AgentIR-4B, please see https://github.com/texttron/AgentIR/tree/main.
 
 ## Citation
 ```