GeorgeIbrahim commited on
Commit
0a274b3
·
1 Parent(s): c08af9f
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -38,7 +38,7 @@ try:
38
  else:
39
  split_values.append("train")
40
 
41
- dataset = dataset.replace_column("split", split_values)
42
  print("Added 'split' column to dataset.")
43
 
44
  else:
 
38
  else:
39
  split_values.append("train")
40
 
41
+ dataset = dataset.add_column("split", split_values)
42
  print("Added 'split' column to dataset.")
43
 
44
  else: