MichaelHuang commited on
Commit
16d4656
1 Parent(s): e2d11fb

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -42,7 +42,7 @@ labels dict = {
42
  from transformers import AutoModelForTokenClassification, AutoTokenizer
43
  import torch
44
 
45
- model = AutoModelForTokenClassification.from_pretrained("MichaelHuang/muril_base_cased_urdu_sentiment")
46
  tokenizer = AutoTokenizer.from_pretrained("google/muril-base-cased")
47
 
48
  def ner_predict(sentence, model, tokenizer):
 
42
  from transformers import AutoModelForTokenClassification, AutoTokenizer
43
  import torch
44
 
45
+ model = AutoModelForTokenClassification.from_pretrained("MichaelHuang/muril_base_cased_hindi_ner")
46
  tokenizer = AutoTokenizer.from_pretrained("google/muril-base-cased")
47
 
48
  def ner_predict(sentence, model, tokenizer):