razent justinphan3110 commited on
Commit
6c6f445
1 Parent(s): 7ecd3e2

Update README.md (#1)

Browse files

- Update README.md (2c59b55b795c26f6f69dd1c0a27d546e598890e2)


Co-authored-by: Long Phan <[email protected]>

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