Spaces:
Runtime error
Runtime error
Update infer.py
Browse files
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 = "
|
| 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
|