Update README.md
Browse files
README.md
CHANGED
@@ -32,8 +32,9 @@ tokenizer = AutoTokenizer.from_pretrained("razent/SciFive-large-Pubmed_PMC-MedNL
|
|
32 |
model = AutoModelForSeq2SeqLM.from_pretrained("razent/SciFive-large-Pubmed_PMC-MedNLI")
|
33 |
model.cuda()
|
34 |
|
35 |
-
|
36 |
-
|
|
|
37 |
|
38 |
encoding = tokenizer.encode_plus(text, padding='max_length', max_length=256, return_tensors="pt")
|
39 |
input_ids, attention_masks = encoding["input_ids"].to("cuda"), encoding["attention_mask"].to("cuda")
|
|
|
32 |
model = AutoModelForSeq2SeqLM.from_pretrained("razent/SciFive-large-Pubmed_PMC-MedNLI")
|
33 |
model.cuda()
|
34 |
|
35 |
+
sent_1 = "In the ED, initial VS revealed T 98.9, HR 73, BP 121/90, RR 15, O2 sat 98% on RA."
|
36 |
+
sent_2 = "The patient is hemodynamically stable"
|
37 |
+
text = f"mednli: sentence1: {sent_1} sentence2: {sent_2}"
|
38 |
|
39 |
encoding = tokenizer.encode_plus(text, padding='max_length', max_length=256, return_tensors="pt")
|
40 |
input_ids, attention_masks = encoding["input_ids"].to("cuda"), encoding["attention_mask"].to("cuda")
|