add two official medical models

SeanLee97 · Aug 1, 2024 · f300c89 · f300c89
1 parent e00ddb5
commit f300c89
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -76,6 +76,8 @@ BERT-based models:
 |----|------|------|------|
 | [WhereIsAI/UAE-Large-V1](https://huggingface.co/WhereIsAI/UAE-Large-V1) | 512 | cls | English, General-purpose |
 | [WhereIsAI/UAE-Code-Large-V1](https://huggingface.co/WhereIsAI/UAE-Code-Large-V1) |  512 | cls | Code Similarity |
+| [WhereIsAI/pubmed-angle-base-en](https://huggingface.co/WhereIsAI/pubmed-angle-base-en) |  512 | cls | Medical Similarity |
+| [WhereIsAI/pubmed-angle-base-en](https://huggingface.co/WhereIsAI/pubmed-angle-large-en) |  512 | cls | Medical Similarity |
 
 LLM-based models:
 

diff --git a/docs/notes/pretrained_models.rst b/docs/notes/pretrained_models.rst
@@ -13,9 +13,15 @@ BERT-based models:
 +------------------------------------+-------------+-------------------+--------------------------+
 | `WhereIsAI/UAE-Code-Large-V1`_     | 512         | cls               | Code Similarity          |
 +------------------------------------+-------------+-------------------+--------------------------+
+| `WhereIsAI/pubmed-angle-base-en`_  | 512         | cls               | Medical Similarity       |
++------------------------------------+-------------+-------------------+--------------------------+
+| `WhereIsAI/pubmed-angle-large-en`_ | 512         | cls               | Medical Similarity       |
++------------------------------------+-------------+-------------------+--------------------------+
 
 .. _WhereIsAI/UAE-Large-V1: https://huggingface.co/WhereIsAI/UAE-Large-V1
 .. _WhereIsAI/UAE-Code-Large-V1: https://huggingface.co/WhereIsAI/UAE-Code-Large-V1
+.. _WhereIsAI/pubmed-angle-base-en: https://huggingface.co/WhereIsAI/pubmed-angle-base-en
+.. _WhereIsAI/pubmed-angle-large-en: https://huggingface.co/WhereIsAI/pubmed-angle-large-en
 
 
 LLM-based models: