Update README.md

README.md

@@ -1,3 +1,9 @@
+---
+language:
+- en
+base_model:
+- BAAI/bge-large-en-v1.5
+---
 
 The bge-large-en-v1.5 model trained on ToolRet-train dataset, a large-scale training dataset for tool retrieval task.
 This trained `ToolRet-trained-bge-large-en-v1.5` can be used to retrieve useful tools for LLM-based agents in downstream tool-use tasks.
@@ -30,4 +36,4 @@ A concrete example in our training dataset.
     "prompt": "Given a `URL availability` task, retrieve tools that check if a given URL is archived and accessible on a specific date in the Wayback Machine."
 }
 ```
-We use the [FlagEmbedding](https://github.com/FlagOpen/FlagEmbedding/) python library to train the embedding models. To train the embedding models on our datasets, please see the `script/train.sh` in our [Github](https://github.com/mangopy/tool-retrieval-benchmark) for more details. The version of FlagEmbedding is `1.3.3` as provided in the `requirements.yml` file.
+We use the [FlagEmbedding](https://github.com/FlagOpen/FlagEmbedding/) python library to train the embedding models. To train the embedding models on our datasets, please see the `script/train.sh` in our [Github](https://github.com/mangopy/tool-retrieval-benchmark) for more details. The version of FlagEmbedding is `1.3.3` as provided in the `requirements.yml` file.