Update README.md

sileod · Jun 8, 2024 · 70a04c5 · 70a04c5
1 parent 9c1280a
commit 70a04c5
Showing 1 changed file with 2 additions and 3 deletions.
diff --git 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
 ```