Update README.md

README.md

@@ -1,3 +1,6 @@
+A small snippet of code is given here in order to retrieve embeddings and gene expression predictions given a DNA, RNA and protein sequence.
+
+```python
 from transformers import AutoTokenizer, AutoModelForMaskedLM
 import numpy as np
 import torch
@@ -29,3 +32,5 @@ torch_output = model.forward(
 
 print(f"Gene expression predictions: {torch_output['gene_expression_predictions']}")
 print(f"Final DNA embedding: {torch_output['final_dna_embeddings']}")
+
+```