Update README.md

README.md

@@ -3,10 +3,8 @@ language:
   - en
 
 tags:
-- token-classification
 - text-classification
-- question-answering
-- text2text-generation
+- mednli
 - text-generation
 
 datasets:
@@ -15,9 +13,12 @@ datasets:
 
 ---
 
-# SciFive Pubmed+PMC Large
+# SciFive Pubmed+PMC Large on MedNLI
 
 ## Introduction
+
+Finetuned SciFive Pubmed+PMC Large model achieved state-of-the-art results on [MedNLI (Medical Natural Language Inference)](https://paperswithcode.com/sota/natural-language-inference-on-mednli)
+
 Paper: [SciFive: a text-to-text transformer model for biomedical literature](https://arxiv.org/abs/2106.03598)
 
 Authors: _Long N. Phan, James T. Anibal, Hieu Tran, Shaurya Chanana, Erol Bahadroglu, Alec Peltekian, Grégoire Altan-Bonnet_
@@ -31,7 +32,7 @@ tokenizer = AutoTokenizer.from_pretrained("razent/SciFive-large-Pubmed_PMC")
 model = AutoModelForSeq2SeqLM.from_pretrained("razent/SciFive-large-Pubmed_PMC")
 ​
 sentence = "Identification of APC2 , a homologue of the adenomatous polyposis coli tumour suppressor ."
-text =  "ncbi_ner: " + sentence + " </s>"
+text =  "mednli: " + sentence + " </s>"
 
 encoding = tokenizer.encode_plus(text, pad_to_max_length=True, return_tensors="pt")
 input_ids, attention_masks = encoding["input_ids"].to("cuda"), encoding["attention_mask"].to("cuda")