diff --git a/README.md b/README.md index e5b243a..cc3598b 100755 --- a/README.md +++ b/README.md @@ -32,9 +32,8 @@ class hparams: learning_rate = 3e-5 # see hf.co/docs/transformers/en/main_classes/trainer#transformers.TrainingArguments tasks = [rte] -model, trainer = tn.Model_Trainer(task,hparams) -trainer.train() -trainer.evaluate() +model, trainer = tn.Model_Trainer(tasks,hparams) +trainer.train(), trainer.evaluate() p = trainer.pipeline() p([{'text':'premise here','text_pair': 'hypothesis here'}]) # HuggingFace pipeline for inference ```