Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -31,7 +31,7 @@ tokenizer = AutoTokenizer.from_pretrained("razent/SciFive-large-Pubmed")
31
  model = AutoModelForSeq2SeqLM.from_pretrained("razent/SciFive-large-Pubmed")
32
 
33
  sentence = "Identification of APC2 , a homologue of the adenomatous polyposis coli tumour suppressor ."
34
- text = "ncbi_ner: " + sentence + " </s>"
35
 
36
  encoding = tokenizer.encode_plus(text, pad_to_max_length=True, return_tensors="pt")
37
  input_ids, attention_masks = encoding["input_ids"].to("cuda"), encoding["attention_mask"].to("cuda")
 
31
  model = AutoModelForSeq2SeqLM.from_pretrained("razent/SciFive-large-Pubmed")
32
 
33
  sentence = "Identification of APC2 , a homologue of the adenomatous polyposis coli tumour suppressor ."
34
+ text = sentence + " </s>"
35
 
36
  encoding = tokenizer.encode_plus(text, pad_to_max_length=True, return_tensors="pt")
37
  input_ids, attention_masks = encoding["input_ids"].to("cuda"), encoding["attention_mask"].to("cuda")