Update README.md

README.md

@@ -17,6 +17,8 @@ extra_gated_fields:
   I agree to use this model for non-commercial use ONLY: checkbox
 ---
 
-# Masked Diffusion Language Model (MDLM) for Protein Sequence Generation
+# MeMDLM: De Novo Membrane Protein Design with Masked Diffusion Language Models
 
-MDLM, introduced by Sahoo et al (arxiv.org/pdf/2406.07524), provides strong generative capabilities to BERT-style models. In this work, we pre-train and fine-tune ESM-2-150M on the MDLM objective to unconditionally generate realistic, high-quality membrane protein sequences.
+![image/png](https://cdn-uploads.huggingface.co/production/uploads/65bbea9a26c639b000501321/uWW6xnJZwQFWDS1QZNQTm.png)
+
+Masked Diffusion Language Models (MDLMs), introduced by Sahoo et al (arxiv.org/pdf/2406.07524), provide strong generative capabilities to BERT-style models. In this work, we pre-train and fine-tune ESM-2-150M on the MDLM objective to unconditionally generate realistic, high-quality membrane protein sequences.