Update README.md

README.md

@@ -54,6 +54,20 @@ pip install git+https://github.com/BiomedSciAI/biomed-multi-alignment.git
 
 A simple example for a task already supported by `ibm/biomed.omics.bl.sm.ma-ted-458m.tcr_epitope_bind`:
 
+from mammal.examples.tcr_epitope_binding.main_infer import load_model, task_infer
+
+tcr_beta_seq = "NAGVTQTPKFQVLKTGQSMTLQCAQDMNHEYMSWYRQDPGMGLRLIHYSVGAGITDQGEVPNGYNVSRSTTEDFPLRLLSAAPSQTSVYFCASSYSWDRVLEQYFGPGTRLTVT"
+epitope_seq = "LLQTGIHVRVSQPSL"
+
+model_inst, tokenizer_op = load_model(device="cpu")
+result = task_infer(
+    model=model_inst,
+    tokenizer_op=tokenizer_op,
+    tcr_beta_seq=tcr_beta_seq,
+    epitope_seq=epitope_seq,
+)
+print(f"The prediction for {epitope_seq} and {tcr_beta_seq} is {result}")
+
 See our detailed example at: on `https://github.com/BiomedSciAI/biomed-multi-alignment`