han7ter commited on
Commit
a769271
·
1 Parent(s): 0e6ce45
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ tokenized_test = test.map(tokenize_function)
30
  # Загружаем предобученную модель
31
  model = AutoModelForSequenceClassification.from_pretrained(
32
  model_name,
33
- num_labels=28)
34
 
35
  # Задаем параметры обучения
36
  training_args = TrainingArguments(
 
30
  # Загружаем предобученную модель
31
  model = AutoModelForSequenceClassification.from_pretrained(
32
  model_name,
33
+ num_labels=29)
34
 
35
  # Задаем параметры обучения
36
  training_args = TrainingArguments(