Update Readme

README.md

@@ -9,7 +9,7 @@ widget:
 ---
 
 ## Token Classification
-
+Classifies Gro's items and metrics
 
 | **tag**                        | **token** |
 |---------------------------------|-----------|
@@ -17,6 +17,7 @@ widget:
 |I-ITEM    | INSIDE ITEM|
 |B-METRIC   |BEGINNING METRIC  |
 |I-METRIC    | INSIDE METRIC|
+|O          | OUTSIDE |
 
 ---
 
@@ -25,71 +26,17 @@ widget:
 The following Flair script was used to train this model: 
 
 ```python
-from flair.data import Corpus
-from flair.datasets import CONLL_2000
-from flair.embeddings import WordEmbeddings, StackedEmbeddings, FlairEmbeddings
-
-# 1. get the corpus
-corpus: Corpus = CONLL_2000()
-
-# 2. what tag do we want to predict?
-tag_type = 'np'
-
-# 3. make the tag dictionary from the corpus
-tag_dictionary = corpus.make_tag_dictionary(tag_type=tag_type)
-
-# 4. initialize each embedding we use
-embedding_types = [
-
-    # contextual string embeddings, forward
-    FlairEmbeddings('news-forward'),
-
-    # contextual string embeddings, backward
-    FlairEmbeddings('news-backward'),
-]
-
-# embedding stack consists of Flair and GloVe embeddings
-embeddings = StackedEmbeddings(embeddings=embedding_types)
-
-# 5. initialize sequence tagger
-from flair.models import SequenceTagger
-
-tagger = SequenceTagger(hidden_size=256,
-                        embeddings=embeddings,
-                        tag_dictionary=tag_dictionary,
-                        tag_type=tag_type)
+from transformers import AutoModelForTokenClassification, AutoTokenizer, pipeline
+tokenizer = AutoTokenizer.from_pretrained("Wanjiru/autotrain_gro_ner")
 
-# 6. initialize trainer
-from flair.trainers import ModelTrainer
+model = AutoModelForTokenClassification.from_pretrained("Wanjiru/autotrain_gro_ner")
 
-trainer = ModelTrainer(tagger, corpus)
-
-# 7. run training
-trainer.train('resources/taggers/chunk-english',
-              train_with_dev=True,
-              max_epochs=150)
+nlp = pipeline("ner", model=model, tokenizer=tokenizer)
+example = "Wanjru"
+ner_res = nlp(example)
+  
 ```
 
 
 
 ---
-
-### Cite
-
-Please cite the following paper when using this model.
-
-```
-@inproceedings{akbik2018coling,
-  title={Contextual String Embeddings for Sequence Labeling},
-  author={Akbik, Alan and Blythe, Duncan and Vollgraf, Roland},
-  booktitle = {{COLING} 2018, 27th International Conference on Computational Linguistics},
-  pages     = {1638--1649},
-  year      = {2018}
-}
-```
-
----
-
-### Issues?
-
-The Flair issue tracker is available [here](https://github.com/flairNLP/flair/issues/).