Update README.md

README.md

@@ -64,8 +64,7 @@ import transformers
 
 
 model = transformers.AutoModel.from_pretrained(
-    'numind/NuNER-v1.0',
-    output_hidden_states=True
+    'numind/NuNER-v1.0'
 )
 tokenizer = transformers.AutoTokenizer.from_pretrained(
     'numind/NuNER-v1.0'
@@ -83,14 +82,7 @@ encoded_input = tokenizer(
 )
 output = model(**encoded_input)
 
-# for better quality
-emb = torch.cat(
-    (output.hidden_states[-1], output.hidden_states[-7]),
-    dim=2
-)
-
-# for better speed
-# emb = output.hidden_states[-1]
+emb = output.last_hidden_state
 ```
 
 ## Citation