smhavens commited on
Commit
d69e74c
·
1 Parent(s): 69ae466
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -58,9 +58,9 @@ def training():
58
  print(dataset)
59
 
60
  train_examples = []
61
- train_data = dataset['train']
62
  # For agility we only 1/2 of our available data
63
- n_examples = dataset['train'].num_rows // 2
64
 
65
  for i in range(n_examples):
66
  example = train_data[i]
 
58
  print(dataset)
59
 
60
  train_examples = []
61
+ train_data = dataset
62
  # For agility we only 1/2 of our available data
63
+ n_examples = dataset.num_rows // 2
64
 
65
  for i in range(n_examples):
66
  example = train_data[i]