razent commited on
Commit
0024494
1 Parent(s): 98daddf

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -2
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
- sentence = "Identification of APC2 , a homologue of the adenomatous polyposis coli tumour suppressor ."
36
- text = "mednli: " + sentence + " </s>"
 
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")