Baidarkaa commited on
Commit
f7feaf3
·
verified ·
1 Parent(s): 597dda2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,13 +6,13 @@ from transformers import pipeline, AutoTokenizer, AutoModelForTokenClassificatio
6
  # ================== КОНФИГУРАЦИЯ ==================
7
  # Можно легко добавить новые модели
8
  MODELS = {
9
- "Davlan/xlm-roberta-base-ner-hrl",
10
  "Babelscape/wikineural-multilingual-ner",
11
  "CAMeL-Lab/bert-base-arabic-camelbert-mix-ner",
12
  }
13
 
14
  # Выбранная модель по умолчанию
15
- DEFAULT_MODEL = "DDavlan/xlm-roberta-base-ner-hrl"
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 = {