kjanh commited on
Commit
b9d24b2
Β·
verified Β·
1 Parent(s): af11ce4

Update infer.py

Browse files
Files changed (1) hide show
  1. infer.py +1 -1
infer.py CHANGED
@@ -8,7 +8,7 @@ LABEL_LIST = ["O", "B-EN", "I-EN"]
8
  LABEL2ID = {l:i for i,l in enumerate(LABEL_LIST)}
9
  ID2LABEL = {i:l for l,i in LABEL2ID.items()}
10
 
11
- model_name = "meandyou200175/detect_english"
12
  model_detect = AutoModelForTokenClassification.from_pretrained(
13
  model_name, num_labels=len(LABEL_LIST),
14
  id2label=ID2LABEL, label2id=LABEL2ID
 
8
  LABEL2ID = {l:i for i,l in enumerate(LABEL_LIST)}
9
  ID2LABEL = {i:l for l,i in LABEL2ID.items()}
10
 
11
+ model_name = "kjanh/detect_english"
12
  model_detect = AutoModelForTokenClassification.from_pretrained(
13
  model_name, num_labels=len(LABEL_LIST),
14
  id2label=ID2LABEL, label2id=LABEL2ID