Update app.py

app.py

@@ -24,12 +24,11 @@ english_ner = pipeline(
 )
 
 # ============================
-# Функция распознавания сущностей и подсчёта метрик
+# Функция распознавания сущностей
 # ============================
-def recognize_entities_auto(text, gold_entities=None):
+def recognize_entities_auto(text):
     """
     text: текст пользователя
-    gold_entities: словарь с эталонными сущностями {'PER': [...], 'ORG': [...], 'LOC': [...]}
     """
     # Определяем язык
     try:
@@ -72,43 +71,17 @@ def recognize_entities_auto(text, gold_entities=None):
         for item in items:
             highlighted.append((item, key))
 
-    # ============================
-    # Подсчёт метрик
-    # ============================
-    metrics_text = ""
-    if gold_entities:
-        for key in ['PER','ORG','LOC']:
-            # Приводим к нижнему регистру для корректного сравнения
-            pred_set = set([p.lower().strip() for p in entities[key]])
-            gold_set = set([g.lower().strip() for g in gold_entities.get(key, [])])
-
-            tp = len(pred_set & gold_set)
-            fp = len(pred_set - gold_set)
-            fn = len(gold_set - pred_set)
-
-            precision = tp / (tp + fp) if (tp + fp) > 0 else 0.0
-            recall = tp / (tp + fn) if (tp + fn) > 0 else 0.0
-            f1 = 2 * precision * recall / (precision + recall) if (precision + recall) > 0 else 0.0
-
-            metrics_text += f"{key}: Precision={precision:.2f}, Recall={recall:.2f}, F1={f1:.2f}\n"
-
-    return highlighted, metrics_text.strip()
+    return highlighted
 
 # ============================
 # Gradio интерфейс
 # ============================
 iface = gr.Interface(
     fn=recognize_entities_auto,
-    inputs=[
-        gr.Textbox(lines=15, placeholder="Введите русский или английский текст здесь..."),
-        gr.JSON(label="Эталонные сущности (опционально)", value={"PER":[],"ORG":[],"LOC":[]})
-    ],
-    outputs=[
-        gr.HighlightedText(label="Выделенные сущности", color_map={"PER":"#faa", "ORG":"#afa", "LOC":"#aaf"}),
-        gr.Textbox(label="Метрики (если указаны эталонные сущности)")
-    ],
+    inputs=gr.Textbox(lines=15, placeholder="Введите русский или английский текст здесь..."),
+    outputs=gr.HighlightedText(label="Выделенные сущности", color_map={"PER":"#faa", "ORG":"#afa", "LOC":"#aaf"}),
     title="Автоматический NER для русского и английского текста",
-    description="PER – человек, ORG – организация, LOC – место. Текст любого языка обрабатывается автоматически. Можно передать эталонные сущности для подсчёта Precision/Recall/F1."
+    description="PER – человек, ORG – организация, LOC – место. Текст любого языка обрабатывается автоматически."
 )
 
 iface.launch()