Update README.md

README.md

@@ -1,3 +1,76 @@
 ---
+language:
+- en
 license: mit
+tags:
+- token-classification
+- entity-recognition
+- foundation-model
+- feature-extraction
+- BERT
+- generic
+datasets:
+- numind/NuNER
+pipeline_tag: token-classification
+inference: false
 ---
+
+# SOTA Entity Recognition English Foundation Model by NuMind 🔥
+
+ This model provides the embedding for the Entity Recognition task in English.
+
+ **We recommend firstly trying [NuNER RoBERTa](https://huggingface.co/numind/NuNER-v0.1) as it usually shows better results**
+
+**Checkout other models by NuMind:**
+* SOTA Multilingual Entity Recognition Foundation Model: [link](https://huggingface.co/numind/entity-recognition-multilingual-general-sota-v1)
+* SOTA Sentiment Analysis Foundation Model: [English](https://huggingface.co/numind/generic-sentiment-v1), [Multilingual](https://huggingface.co/numind/generic-sentiment-multi-v1)
+
+## About
+
+[bert-base-uncased](https://huggingface.co/google-bert/bert-base-uncased) fine-tuned on [NuNER data](https://huggingface.co/datasets/numind/NuNER).
+
+**Metrics:**
+
+Read more about evaluation protocol & datasets in our [paper](https://arxiv.org/abs/2402.15343) and [blog post](https://www.numind.ai/blog/a-foundation-model-for-entity-recognition).
+
+## Usage
+
+Embeddings can be used out of the box or fine-tuned on specific datasets. 
+
+Get embeddings:
+
+
+```python
+import torch
+import transformers
+
+
+model = transformers.AutoModel.from_pretrained(
+    'numind/NuNER-BERT-v1.0',
+    output_hidden_states=True
+)
+tokenizer = transformers.AutoTokenizer.from_pretrained(
+    'numind/NuNER-BERT-v1.0'
+)
+
+text = [
+    "NuMind is an AI company based in Paris and USA.",
+    "See other models from us on https://huggingface.co/numind"
+]
+encoded_input = tokenizer(
+    text,
+    return_tensors='pt',
+    padding=True,
+    truncation=True
+)
+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]
+```