PENIS4

app.py

@@ -12,6 +12,7 @@ model_name = "DeepPavlov/rubert-base-cased"
 # 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'}
 df = pd.read_parquet("hf://datasets/mteb/RuSciBenchOECDClassification/" + splits["train"])
+df = df.head(500)
 
 # Конвертируем датафрейм в Dataset
 train, test = train_test_split(df, test_size=0.2)