Update README.md

README.md

@@ -19,8 +19,8 @@ mgen test --model SequenceClassification --model.backbone aido_protein_16b_v1 --
 ```python
 from modelgenerator.tasks import Embed
 model = Embed.from_config({"model.backbone": "aido_protein_16b_v1"}).eval()
-collated_batch = model.collate({"sequences": ["HELLQ", "WRLD"]})
-embedding = model(collated_batch)
+transformed_batch = model.transform({"sequences": ["HELLQ", "WRLD"]})
+embedding = model(transformed_batch)
 print(embedding.shape)
 print(embedding)
 ```
@@ -29,8 +29,8 @@ print(embedding)
 import torch
 from modelgenerator.tasks import SequenceClassification
 model = SequenceClassification.from_config({"model.backbone": "aido_protein_16b_v1", "model.n_classes": 2}).eval()
-collated_batch = model.collate({"sequences": ["HELLQ", "WRLD"]})
-logits = model(collated_batch)
+transformed_batch = model.transform({"sequences": ["HELLQ", "WRLD"]})
+logits = model(transformed_batch)
 print(logits)
 print(torch.argmax(logits, dim=-1))
 ```
@@ -39,8 +39,8 @@ print(torch.argmax(logits, dim=-1))
 import torch
 from modelgenerator.tasks import TokenClassification
 model = TokenClassification.from_config({"model.backbone": "aido_protein_16b_v1", "model.n_classes": 3}).eval()
-collated_batch = model.collate({"sequences": ["HELLQ", "WRLD"]})
-logits = model(collated_batch)
+transformed_batch = model.transform({"sequences": ["HELLQ", "WRLD"]})
+logits = model(transformed_batch)
 print(logits)
 print(torch.argmax(logits, dim=-1))
 ```
@@ -48,8 +48,8 @@ print(torch.argmax(logits, dim=-1))
 ```python
 from modelgenerator.tasks import SequenceRegression
 model = SequenceRegression.from_config({"model.backbone": "aido_protein_16b_v1"}).eval()
-collated_batch = model.collate({"sequences": ["HELLQ", "WRLD"]})
-logits = model(collated_batch)
+transformed_batch = model.transform({"sequences": ["HELLQ", "WRLD"]})
+logits = model(transformed_batch)
 print(logits)
 ```