Berbex commited on
Commit
44aeeb4
Β·
1 Parent(s): 418f0d9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -33,10 +33,10 @@ def preprocess_data(examples):
33
 
34
  encoded_dataset = dataset.map(preprocess_data, batched=True, remove_columns=dataset['train'].column_names)
35
 
36
- example = encoded_dataset['train'][0]
37
  console.log(example['labels'])
38
 
39
-
40
 
41
 
42
 
 
33
 
34
  encoded_dataset = dataset.map(preprocess_data, batched=True, remove_columns=dataset['train'].column_names)
35
 
36
+ example = encoded_dataset['train'][3000]
37
  console.log(example['labels'])
38
 
39
+ encoded_dataset.set_format("torch")
40
 
41
 
42