Update app.py
Browse files
app.py
CHANGED
|
@@ -6,13 +6,13 @@ from transformers import pipeline, AutoTokenizer, AutoModelForTokenClassificatio
|
|
| 6 |
# ================== КОНФИГУРАЦИЯ ==================
|
| 7 |
# Можно легко добавить новые модели
|
| 8 |
MODELS = {
|
| 9 |
-
"
|
| 10 |
"Babelscape/wikineural-multilingual-ner",
|
| 11 |
"CAMeL-Lab/bert-base-arabic-camelbert-mix-ner",
|
| 12 |
}
|
| 13 |
|
| 14 |
# Выбранная модель по умолчанию
|
| 15 |
-
DEFAULT_MODEL = "
|
| 16 |
|
| 17 |
# Цвета для разных типов сущностей (для красивого отображения)
|
| 18 |
ENTITY_COLORS = {
|
|
|
|
| 6 |
# ================== КОНФИГУРАЦИЯ ==================
|
| 7 |
# Можно легко добавить новые модели
|
| 8 |
MODELS = {
|
| 9 |
+
"Leo97/KoELECTRA-small-v3-modu-ner",
|
| 10 |
"Babelscape/wikineural-multilingual-ner",
|
| 11 |
"CAMeL-Lab/bert-base-arabic-camelbert-mix-ner",
|
| 12 |
}
|
| 13 |
|
| 14 |
# Выбранная модель по умолчанию
|
| 15 |
+
DEFAULT_MODEL = "Leo97/KoELECTRA-small-v3-modu-ner"
|
| 16 |
|
| 17 |
# Цвета для разных типов сущностей (для красивого отображения)
|
| 18 |
ENTITY_COLORS = {
|