AnanthanarayananSeetharaman
commited on
Commit
•
966c992
1
Parent(s):
e1a62fc
Update README.md
Browse filesupdated the model name and model card in yaml format
README.md
CHANGED
@@ -11,7 +11,7 @@ metrics:
|
|
11 |
- precision
|
12 |
- recall
|
13 |
- f1
|
14 |
-
model_name:
|
15 |
widget:
|
16 |
- text: "The SALES of BEER and WINE in TTL US is increasing."
|
17 |
---
|
@@ -34,8 +34,8 @@ The model is trained to recognize the following entities:
|
|
34 |
from transformers import AutoTokenizer, AutoModelForTokenClassification, pipeline
|
35 |
|
36 |
# Load the model and tokenizer
|
37 |
-
model = AutoModelForTokenClassification.from_pretrained("
|
38 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
39 |
|
40 |
# Create a pipeline
|
41 |
ner_pipeline = pipeline("ner", model=model, tokenizer=tokenizer)
|
|
|
11 |
- precision
|
12 |
- recall
|
13 |
- f1
|
14 |
+
model_name: osa-custom-ner-model
|
15 |
widget:
|
16 |
- text: "The SALES of BEER and WINE in TTL US is increasing."
|
17 |
---
|
|
|
34 |
from transformers import AutoTokenizer, AutoModelForTokenClassification, pipeline
|
35 |
|
36 |
# Load the model and tokenizer
|
37 |
+
model = AutoModelForTokenClassification.from_pretrained("AnanthanarayananSeetharaman/osa-custom-ner-model")
|
38 |
+
tokenizer = AutoTokenizer.from_pretrained("AnanthanarayananSeetharaman/osa-custom-ner-model")
|
39 |
|
40 |
# Create a pipeline
|
41 |
ner_pipeline = pipeline("ner", model=model, tokenizer=tokenizer)
|