Asanaly commited on
Commit
b3123ac
·
verified ·
1 Parent(s): f95a6bb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -3,10 +3,10 @@ from transformers import pipeline
3
  from langdetect import detect
4
 
5
  # ============================
6
- # Модели
7
  # ============================
8
  MODELS = {
9
- "Russian": "DeepPavlov/ner_rubert",
10
  "English": "dbmdz/bert-large-cased-finetuned-conll03-english"
11
  }
12
 
@@ -51,7 +51,6 @@ def auto_ner(text):
51
  result_text = ""
52
  for key, values in output.items():
53
  if values:
54
- # Убираем дубли и соединяем через ;
55
  result_text += f"{key}: {'; '.join(list(dict.fromkeys(values)))}\n"
56
  return result_text.strip()
57
 
@@ -60,7 +59,7 @@ def auto_ner(text):
60
  # ============================
61
  iface = gr.Interface(
62
  fn=auto_ner,
63
- inputs=gr.Textbox(lines=15, placeholder="Введите текст здесь (русский или английский, можно несколько предложений)..."),
64
  outputs=gr.Textbox(label="Распознанные сущности (PER/ORG/LOC)"),
65
  title="Автоматический NER для русского и английского текста",
66
  description="PER – человек, ORG – организация, LOC – место. Текст любого языка обрабатывается автоматически."
 
3
  from langdetect import detect
4
 
5
  # ============================
6
+ # Публичные модели
7
  # ============================
8
  MODELS = {
9
+ "Russian": "blinoff/bert-base-russian-ner",
10
  "English": "dbmdz/bert-large-cased-finetuned-conll03-english"
11
  }
12
 
 
51
  result_text = ""
52
  for key, values in output.items():
53
  if values:
 
54
  result_text += f"{key}: {'; '.join(list(dict.fromkeys(values)))}\n"
55
  return result_text.strip()
56
 
 
59
  # ============================
60
  iface = gr.Interface(
61
  fn=auto_ner,
62
+ inputs=gr.Textbox(lines=15, placeholder="Введите текст здесь (русский или английский, несколько предложений)..."),
63
  outputs=gr.Textbox(label="Распознанные сущности (PER/ORG/LOC)"),
64
  title="Автоматический NER для русского и английского текста",
65
  description="PER – человек, ORG – организация, LOC – место. Текст любого языка обрабатывается автоматически."