too much roooooooows

app.py

@@ -15,7 +15,7 @@ model_name = "cointegrated/rubert-tiny2"
 # Login using e.g. `huggingface-cli login` to access this dataset
 splits = {'train': 'train.json', 'test': 'test.json'}
 df = pd.read_json("hf://datasets/Den4ikAI/gibberish_dataset/" + splits["train"])
-df.head(500)
+df = df.head(500)
 
 # Конвертируем датафрейм в Dataset
 train, test = train_test_split(df, test_size=0.2)