Berbex commited on
Commit
1b13b28
β€’
1 Parent(s): 624a2c3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -1
app.py CHANGED
@@ -95,7 +95,18 @@ def compute_metrics(p: EvalPrediction):
95
  return result
96
 
97
 
98
- console.log(encoded_dataset['train']['input_ids'][0])
 
 
 
 
 
 
 
 
 
 
 
99
 
100
  """
101
 
 
95
  return result
96
 
97
 
98
+ trainer = Trainer(
99
+ model,
100
+ args,
101
+ train_dataset=encoded_dataset["train"],
102
+ eval_dataset=encoded_dataset["validation"],
103
+ tokenizer=tokenizer,
104
+ compute_metrics=compute_metrics
105
+ )
106
+
107
+ trainer.train()
108
+
109
+ trainer.evaluate()
110
 
111
  """
112