razent
/

SciFive-base-Pubmed

Text Classification

text2text-generation

token-classification

question-answering

text-generation

text-generation-inference

Model card Files Files and versions

Update README.md

#1

by justinphan3110 - opened Nov 26, 2022

base: refs/heads/main

←

from: refs/pr/1

Discussion Files changed

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -30,7 +30,7 @@ tokenizer = AutoTokenizer.from_pretrained("razent/SciFive-base-Pubmed")
 model = AutoModelForSeq2SeqLM.from_pretrained("razent/SciFive-base-Pubmed")
 sentence = "Identification of APC2 , a homologue of the adenomatous polyposis coli tumour suppressor ."
-text =  "ncbi_ner: " + 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")

 model = AutoModelForSeq2SeqLM.from_pretrained("razent/SciFive-base-Pubmed")
 sentence = "Identification of APC2 , a homologue of the adenomatous polyposis coli tumour suppressor ."
+text = 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")