Update README.md

README.md

@@ -41,7 +41,7 @@ mgen test --model SequenceClassification --model.backbone aido_rna_1b600m --data
 #### Embedding
 ```python
 from modelgenerator.tasks import Embed
-model = Embed.from_config({"model.backbone": "aido_dna_7b"}).eval()
+model = Embed.from_config({"model.backbone": "aido_rna_1b600m"}).eval()
 collated_batch = model.collate({"sequences": ["ACGT", "AGCT"]})
 embedding = model(collated_batch)
 print(embedding.shape)
@@ -51,7 +51,7 @@ print(embedding)
 ```python
 import torch
 from modelgenerator.tasks import SequenceClassification
-model = SequenceClassification.from_config({"model.backbone": "aido_dna_7b", "model.n_classes": 2}).eval()
+model = SequenceClassification.from_config({"model.backbone": "aido_rna_1b600m", "model.n_classes": 2}).eval()
 collated_batch = model.collate({"sequences": ["ACGT", "AGCT"]})
 logits = model(collated_batch)
 print(logits)
@@ -61,7 +61,7 @@ print(torch.argmax(logits, dim=-1))
 ```python
 import torch
 from modelgenerator.tasks import TokenClassification
-model = TokenClassification.from_config({"model.backbone": "aido_dna_7b", "model.n_classes": 3}).eval()
+model = TokenClassification.from_config({"model.backbone": "aido_rna_1b600m", "model.n_classes": 3}).eval()
 collated_batch = model.collate({"sequences": ["ACGT", "AGCT"]})
 logits = model(collated_batch)
 print(logits)
@@ -70,10 +70,11 @@ print(torch.argmax(logits, dim=-1))
 #### Sequence-level Regression
 ```python
 from modelgenerator.tasks import SequenceRegression
-model = SequenceRegression.from_config({"model.backbone": "aido_dna_7b"}).eval()
+model = SequenceRegression.from_config({"model.backbone": "aido_rna_1b600m"}).eval()
 collated_batch = model.collate({"sequences": ["ACGT", "AGCT"]})
 logits = model(collated_batch)
 print(logits)
+```
 
 ### Get RNA sequence embedding
 ```python