marry1908 commited on
Commit
8e56d7c
·
verified ·
1 Parent(s): 8b8c886

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -16,14 +16,14 @@ except ImportError:
16
  history = []
17
  MAX_HISTORY = 10
18
 
19
- # Предопределённые темы для классификации
20
  PREDEFINED_TOPICS = [
21
  "спорт", "политика", "наука", "технологии",
22
  "здоровье", "образование", "культура", "экономика",
23
  "путешествия", "развлечения", "погода", "происшествия"
24
  ]
25
 
26
- # Словарь для расшифровки тем на русский с эмодзи
27
  TOPIC_DESCRIPTIONS = {
28
  "спорт": "🏀 Спорт",
29
  "политика": "🏛️ Политика",
@@ -39,7 +39,7 @@ TOPIC_DESCRIPTIONS = {
39
  "происшествия": "🚨 Происшествия"
40
  }
41
 
42
- # Простая эвристическая классификация (запасной вариант)
43
  def simple_classifier(text, topics):
44
  """Простой классификатор на основе ключевых слов"""
45
  text_lower = text.lower()
@@ -87,7 +87,6 @@ def classify_topic(text, show_top_n=3):
87
  # Пытаемся использовать трансформеры, если доступны
88
  if transformers_available:
89
  try:
90
- # ПРОБУЕМ ЛЁГКУЮ МОДЕЛЬ ДЛЯ РУССКОГО ЯЗЫКА
91
  classifier = pipeline("zero-shot-classification",
92
  model="MoritzLaurer/mDeBERTa-v3-base-mnli-xnli",
93
  device=-1)
 
16
  history = []
17
  MAX_HISTORY = 10
18
 
19
+ # Темы для классификации
20
  PREDEFINED_TOPICS = [
21
  "спорт", "политика", "наука", "технологии",
22
  "здоровье", "образование", "культура", "экономика",
23
  "путешествия", "развлечения", "погода", "происшествия"
24
  ]
25
 
26
+
27
  TOPIC_DESCRIPTIONS = {
28
  "спорт": "🏀 Спорт",
29
  "политика": "🏛️ Политика",
 
39
  "происшествия": "🚨 Происшествия"
40
  }
41
 
42
+
43
  def simple_classifier(text, topics):
44
  """Простой классификатор на основе ключевых слов"""
45
  text_lower = text.lower()
 
87
  # Пытаемся использовать трансформеры, если доступны
88
  if transformers_available:
89
  try:
 
90
  classifier = pipeline("zero-shot-classification",
91
  model="MoritzLaurer/mDeBERTa-v3-base-mnli-xnli",
92
  device=-1)