Update README.md

README.md

@@ -45,7 +45,7 @@ To create embeddings and for classification:
 title = 'MIReAD: simple method for learning scientific representations'
 abstract = 'Learning semantically meaningful representations from scientific documents can ...'
 text = title + tokenizer.sep_token + abstract
-tokens = tokenizer(sents,
+tokens = tokenizer(text,
                    max_length=512,
                    padding=True,
                    truncation=True,
@@ -54,12 +54,12 @@ tokens = tokenizer(sents,
 
 # for embeddings
 with torch.no_grad():
-  out = model.bert(**tokens)
-  feature = out.last_hidden_state[:, 0, :]
+  output = model.bert(**tokens)
+  embedding = output.last_hidden_state[:, 0, :]
 
 # for classification
-out = model(**inputs)
-logits = out.logits 
+output = model(**tokens)
+class = out.logits 
 ```
 
 ## Training procedure