ASureevaA commited on
Commit
2717a3f
·
1 Parent(s): b2f68cd
Files changed (1) hide show
  1. app.py +8 -2
app.py CHANGED
@@ -72,12 +72,19 @@ def run_text_classification(input_text: str) -> str:
72
  Анализ текста трансформером:
73
  используем sentiment-классификатор как пример.
74
  Возвращаем строку вида: "POSITIVE (score=0.982)".
 
 
 
75
  """
76
  cleaned_text: str = input_text.strip()
77
  if not cleaned_text:
78
  return ""
79
 
80
- classifier_result_list = text_classifier_pipeline(cleaned_text)
 
 
 
 
81
  classifier_result = classifier_result_list[0]
82
 
83
  label_value: str = str(classifier_result.get("label", ""))
@@ -86,7 +93,6 @@ def run_text_classification(input_text: str) -> str:
86
  classification_text: str = f"{label_value} (score={score_value:.3f})"
87
  return classification_text
88
 
89
-
90
  # ============================
91
  # 4. Трансформер #2: суммаризация (английский)
92
  # ============================
 
72
  Анализ текста трансформером:
73
  используем sentiment-классификатор как пример.
74
  Возвращаем строку вида: "POSITIVE (score=0.982)".
75
+
76
+ ВАЖНО: жёстко ограничиваем длину до 512 токенов, чтобы не ловить
77
+ ошибку DistilBERT по max_position_embeddings.
78
  """
79
  cleaned_text: str = input_text.strip()
80
  if not cleaned_text:
81
  return ""
82
 
83
+ classifier_result_list = text_classifier_pipeline(
84
+ cleaned_text,
85
+ truncation=True,
86
+ max_length=512,
87
+ )
88
  classifier_result = classifier_result_list[0]
89
 
90
  label_value: str = str(classifier_result.get("label", ""))
 
93
  classification_text: str = f"{label_value} (score={score_value:.3f})"
94
  return classification_text
95
 
 
96
  # ============================
97
  # 4. Трансформер #2: суммаризация (английский)
98
  # ============================