Update README.md

README.md

@@ -43,13 +43,40 @@ tokenized = tokenizer(text, return_tensors='pt')
 output = model(**tokenized)
 
 # result
-pred = np.argmax(output[0].cpu().detach().numpy(), axis=2)[0][1:-1]
+preds = np.argmax(output[0].cpu().detach().numpy(), axis=2)[0][1:-1]
 
-# check pred
-for txt, pred in zip(tokenizer.tokenize(text), pred):
+# check preds
+for txt, pred in zip(tokenizer.tokenize(text), preds):
     print("{}\t{}".format(id2tag[pred], txt))
     # B_MT mental 
-    # B_MT disorder 
+    # B_MT disorder
+    # O	can
+    # O	also
+    # O	contribute
+    # O	to
+    # O	the
+    # B_MT	development
+    # O	of
+    # B_MT	diabetes
+    # O	through
+    # O	various
+    # B_MT	mechanism
+    # O	including
+    # O	increased
+    # B_MT	stress
+    # O	,
+    # O	poor
+    # B_MT	self
+    # B_MT	care
+    # B_MT	behavior
+    # O	,
+    # O	and
+    # B_MT	adverse
+    # I_MT	effect
+    # O	on
+    # B_MT	glucose
+    # B_MT	metabolism
+    # O	.
 ```
 
 ## Framework versions