Update README.md

README.md

@@ -7,9 +7,9 @@ tags:
 - transformers
 ---
 
-# {MODEL_NAME}
+# pritamdeka/SapBERT-mnli-snli-scinli-scitail-mednli-stsb
 
-This is a [sentence-transformers](https://www.SBERT.net) model: It maps sentences & paragraphs to a 768 dimensional dense vector space and can be used for tasks like clustering or semantic search.
+This is a [sentence-transformers](https://www.SBERT.net) model: It maps sentences & paragraphs to a 768 dimensional dense vector space and can be used for tasks like clustering or semantic search. It has been trained over the SNLI, MNLI, SCINLI, SCITAIL, MEDNLI and STSB datasets for providing robust sentence embeddings.
 
 <!--- Describe your model here -->
 
@@ -27,7 +27,7 @@ Then you can use the model like this:
 from sentence_transformers import SentenceTransformer
 sentences = ["This is an example sentence", "Each sentence is converted"]
 
-model = SentenceTransformer('{MODEL_NAME}')
+model = SentenceTransformer('pritamdeka/SapBERT-mnli-snli-scinli-scitail-mednli-stsb')
 embeddings = model.encode(sentences)
 print(embeddings)
 ```
@@ -53,8 +53,8 @@ def mean_pooling(model_output, attention_mask):
 sentences = ['This is an example sentence', 'Each sentence is converted']
 
 # Load model from HuggingFace Hub
-tokenizer = AutoTokenizer.from_pretrained('{MODEL_NAME}')
-model = AutoModel.from_pretrained('{MODEL_NAME}')
+tokenizer = AutoTokenizer.from_pretrained('pritamdeka/SapBERT-mnli-snli-scinli-scitail-mednli-stsb')
+model = AutoModel.from_pretrained('pritamdeka/SapBERT-mnli-snli-scinli-scitail-mednli-stsb')
 
 # Tokenize sentences
 encoded_input = tokenizer(sentences, padding=True, truncation=True, return_tensors='pt')
@@ -122,4 +122,17 @@ SentenceTransformer(
 
 ## Citing & Authors
 
-<!--- Describe where people can find more information -->
+<!--- Describe where people can find more information -->
+
+If you use the model kindly cite the following work
+
+```
+@inproceedings{deka2022evidence,
+  title={Evidence Extraction to Validate Medical Claims in Fake News Detection},
+  author={Deka, Pritam and Jurek-Loughrey, Anna and others},
+  booktitle={International Conference on Health Information Science},
+  pages={3--15},
+  year={2022},
+  organization={Springer}
+}
+```