Spaces:
Build error
Build error
Update flair_recognizer.py
Browse files- flair_recognizer.py +5 -8
flair_recognizer.py
CHANGED
@@ -78,17 +78,13 @@ class FlairRecognizer(EntityRecognizer):
|
|
78 |
({"AGE"}, {"AGE"}),
|
79 |
({"CURRENCY"}, {"CURRENCY"}),
|
80 |
({"SWIFT_CODE"}, {"SWIFT_CODE"}),
|
81 |
-
|
82 |
-
|
83 |
-
|
84 |
-
# ({"US_DRIVER_LICENSE"}, {"US_DRIVER_LICENSE"}),
|
85 |
]
|
86 |
|
87 |
MODEL_LANGUAGES = {
|
88 |
"en": "beki/flair-ner-debug-english",
|
89 |
-
# "es": "flair/ner-spanish-large",
|
90 |
-
# "de": "flair/ner-german-large",
|
91 |
-
# "nl": "flair/ner-dutch-large",
|
92 |
}
|
93 |
|
94 |
PRESIDIO_EQUIVALENCES = {
|
@@ -102,7 +98,8 @@ class FlairRecognizer(EntityRecognizer):
|
|
102 |
"IBAN_CODE": "IBAN_CODE",
|
103 |
"IP_ADDRESS": "IP_ADDRESS",
|
104 |
"EMAIL_ADDRESS": "EMAIL",
|
105 |
-
|
|
|
106 |
}
|
107 |
|
108 |
def __init__(
|
|
|
78 |
({"AGE"}, {"AGE"}),
|
79 |
({"CURRENCY"}, {"CURRENCY"}),
|
80 |
({"SWIFT_CODE"}, {"SWIFT_CODE"}),
|
81 |
+
({"US_ITIN"}, {"US_ITIN"}),
|
82 |
+
({"US_BANK_NUMBER"}, {"US_BANK_NUMBER"}),
|
83 |
+
({"US_DRIVER_LICENSE"}, {"US_DRIVER_LICENSE"}),
|
|
|
84 |
]
|
85 |
|
86 |
MODEL_LANGUAGES = {
|
87 |
"en": "beki/flair-ner-debug-english",
|
|
|
|
|
|
|
88 |
}
|
89 |
|
90 |
PRESIDIO_EQUIVALENCES = {
|
|
|
98 |
"IBAN_CODE": "IBAN_CODE",
|
99 |
"IP_ADDRESS": "IP_ADDRESS",
|
100 |
"EMAIL_ADDRESS": "EMAIL",
|
101 |
+
"US_DRIVER_LICENSE": "US_DRIVER_LICENSE",
|
102 |
+
"US_BANK_NUMBER": "US_BANK_NUMBER",
|
103 |
}
|
104 |
|
105 |
def __init__(
|