russian-ner / config.py
dnkdm's picture
Fix: use working Russian NER model
45ae6eb
raw
history blame contribute delete
913 Bytes
"""
Конфигурация и константы для NER-приложения.
"""
# Лимиты
MAX_CHARS = 2000
MAX_BATCH_ROWS = 100
HISTORY_SIZE = 10
# Доступные модели NER
MODELS = {
"WikiNEuRal (multilingual)": "Babelscape/wikineural-multilingual-ner",
"BERT Russian NER": "Gherman/bert-base-NER-Russian"
}
# Цветовая схема для подсветки сущностей
COLOR_MAP = {
"PER": "#3b82f6", # Синий — персоны
"ORG": "#22c55e", # Зелёный — организации
"LOC": "#f97316", # Оранжевый — локации
"MISC": "#a855f7" # Фиолетовый — прочее
}
# Описания типов сущностей
ENTITY_LABELS = {
"PER": "Персона (ФИО)",
"ORG": "Организация",
"LOC": "Локация (место)",
"MISC": "Прочее"
}