Change model

app.py

@@ -7,7 +7,7 @@ from sklearn.model_selection import train_test_split
 from transformers import (AutoTokenizer, AutoModelForSequenceClassification,
                           TrainingArguments, Trainer)
 
-model_name = "DeepPavlov/rubert-base-cased"
+model_name = "cointegrated/rubert-tiny2"
 
 # Login using e.g. `huggingface-cli login` to access this dataset
 splits = {'train': 'data/train-00000-of-00001.parquet', 'test': 'data/test-00000-of-00001.parquet'}