justinphan3110 commited on
Commit
7ff7dbe
1 Parent(s): d6e6df3

Update README.md

Browse files
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_PMC")
31
  model = AutoModelForSeq2SeqLM.from_pretrained("razent/SciFive-large-Pubmed_PMC")
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_PMC")
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")