GeorgeIbrahim commited on
Commit
81c4d52
·
1 Parent(s): 77dd516
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -40,7 +40,7 @@ try:
40
  split_values.append("train")
41
 
42
  # Add 'split' column to the dataset
43
- dataset = dataset.add_column("split", split_values)
44
  print("Added 'split' column to dataset.")
45
  else:
46
  print("'split' column already exists.")
 
40
  split_values.append("train")
41
 
42
  # Add 'split' column to the dataset
43
+ dataset = dataset.replace_column("split", split_values)
44
  print("Added 'split' column to dataset.")
45
  else:
46
  print("'split' column already exists.")