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__(
|