Update README.md

README.md

@@ -1,10 +1,15 @@
 ---
-datasets:
-- pubmed
 language:
 - en
 tags:
 - BERT
+- medical
+pipeline_tag: token-classification
+widget:
+- text: 63 year old woman with history of CAD presented to ER
+  example_title: Example-1
+- text: 63 year old woman diagnosed with CAD
+  example_title: Example-2
 ---
 # Model Card for Model ID
 
@@ -50,4 +55,11 @@ for txt, pred in zip(tokenizer.tokenize(text), pred):
     print("{}\t{}".format(id2tag[pred], txt))
     # B_MT mental 
     # B_MT disorder 
-```
+```
+
+## Framework versions
+- transformers : 4.39.1
+- torch : 2.1.0+cu121
+- datasets : 2.18.0
+- tokenizers : 0.15.2
+- numpy : 1.20.0