Update README.md

README.md

@@ -59,7 +59,7 @@ with torch.no_grad():
 
 # for classification
 output = model(**tokens)
-class = out.logits 
+class = output.logits 
 ```
 
 ## Training procedure