Commit
•
611f98f
1
Parent(s):
d6e6df3
Update README.md (#1)
Browse files- Update README.md (7ff7dbe32a60e26cfe09ccbe37b03b718b449517)
Co-authored-by: Long Phan <[email protected]>
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 =
|
| 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")
|