AnanthanarayananSeetharaman commited on
Commit
966c992
1 Parent(s): e1a62fc

Update README.md

Browse files

updated the model name and model card in yaml format

Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -11,7 +11,7 @@ metrics:
11
  - precision
12
  - recall
13
  - f1
14
- model_name: my-custom-ner-model
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("username/my-custom-ner-model")
38
- tokenizer = AutoTokenizer.from_pretrained("username/my-custom-ner-model")
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)