MichaelHuang
commited on
Commit
•
16d4656
1
Parent(s):
e2d11fb
Update README.md
Browse files
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/
|
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):
|