smhavens commited on
Commit
940deb0
·
1 Parent(s): fe5d9d2

test again

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -63,11 +63,11 @@ def training():
63
  # print(example)
64
  train_examples.append(InputExample(texts=[example['text'], example['label']]))
65
 
66
- # train_dataloader = DataLoader(train_examples, shuffle=True, batch_size=16)
67
 
68
 
69
 
70
- embeddings = finetune(train_examples)
71
 
72
  return (dataset['train'].num_rows, type(dataset['train'][0]), type(dataset['train'][0]['text']), dataset['train'][0], embeddings)
73
 
 
63
  # print(example)
64
  train_examples.append(InputExample(texts=[example['text'], example['label']]))
65
 
66
+ train_dataloader = DataLoader(train_examples, shuffle=True, batch_size=16)
67
 
68
 
69
 
70
+ embeddings = finetune(train_dataloader)
71
 
72
  return (dataset['train'].num_rows, type(dataset['train'][0]), type(dataset['train'][0]['text']), dataset['train'][0], embeddings)
73