justinphan3110
commited on
Commit
•
2c59b55
1
Parent(s):
7ecd3e2
Update README.md
Browse files
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 =
|
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")
|