Update README.md

README.md

@@ -1,9 +1,18 @@
 ---
-language: en
-
+license: apache-2.0
+language:
+- en
 tags:
 - biomedical
-- lexical-semantics
+- lexical semantics
+- bionlp
+- biology
+- science
+- embedding
+- entity linking
+---
+---
+
 
 datasets:
 - UMLS
@@ -11,8 +20,11 @@ datasets:
 **[news]** A cross-lingual extension of SapBERT will appear in the main onference of **ACL 2021**! <br>
 **[news]** SapBERT will appear in the conference proceedings of **NAACL 2021**!
 
+### Expected input and output
+The input should be a string of biomedical entity names, e.g., "covid infection" or "Hydroxychloroquine". The [CLS] embedding of the last layer is regarded as the output.
+
 ### SapBERT-PubMedBERT
-SapBERT by [Liu et al. (2020)](https://arxiv.org/pdf/2010.11784.pdf). Trained with [UMLS](https://www.nlm.nih.gov/research/umls/licensedcontent/umlsknowledgesources.html) 2020AA (English only), using [microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract-fulltext](https://huggingface.co/microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract-fulltext) as the base model. Please use [CLS] as the representation of the input.
+SapBERT by [Liu et al. (2020)](https://arxiv.org/pdf/2010.11784.pdf). Trained with [UMLS](https://www.nlm.nih.gov/research/umls/licensedcontent/umlsknowledgesources.html) 2020AA (English only), using [microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract-fulltext](https://huggingface.co/microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract-fulltext) as the base model.
 
 ### Citation
 ```bibtex