Update README.md

README.md

@@ -32,7 +32,7 @@ The pre-training data contains 42 million unique ncRNA sequences from RNAcentral
 Build any downstream models from this backbone
 
 ### Get RNA sequence embedding
-```
+```python
 from genbio_finetune.tasks import Embed
 model = Embed.from_config({"model.backbone": "rnafm"}).eval()
 collated_batch = model.collate({"sequences": ["ACGT", "ACGT"]})
@@ -42,7 +42,7 @@ print(embedding)
 ```
 
 ### Sequence-level classification
-```
+```python
 import torch
 from genbio_finetune.tasks import SequenceClassification
 model = SequenceClassification.from_config({"model.backbone": "rnafm", "model.n_classes": 2}).eval()
@@ -53,7 +53,7 @@ print(torch.argmax(logits, dim=-1))
 ```
 
 ### Token-level classification
-```
+```python
 import torch
 from genbio_finetune.tasks import TokenClassification
 model = TokenClassification.from_config({"model.backbone": "rnafm", "model.n_classes": 3}).eval()
@@ -69,7 +69,7 @@ print(torch.argmax(logits, dim=-1))
 
 
 ### Sequence-level regression
-```
+```python
 from genbio_finetune.tasks import SequenceRegression
 model = SequenceRegression.from_config({"model.backbone": "rnafm"}).eval()
 collated_batch = model.collate({"sequences": ["ACGT", "AGCT"]})