Update README.md

README.md

@@ -6,8 +6,6 @@ tags:
 
 # METL
 
-<!-- Provide a quick summary of what the model is/does. -->
-
 Mutational Effect Transfer Learning (METL) is a framework for pretraining and finetuning biophysics-informed protein language models. 
 
 
@@ -17,15 +15,15 @@ This 🤗 repository contains a wrapper meant to facilitate the ease of use of M
 
 ### Model Description
 
-METL is discussed in the (paper)[https://www.biorxiv.org/content/10.1101/2024.03.15.585128v1] in further detail.
+METL is discussed in the (paper)[https://www.biorxiv.org/content/10.1101/2024.03.15.585128v1] in further detail. The github contains more documentation and includes scripts for training and predicting with METL. A google colab notebook for finetuning and predicting on publically available METL models is made available as well [here](https://github.com/gitter-lab/metl/tree/main/notebooks).
 
 ### Model Sources [optional]
 
 <!-- Provide the basic links for the model. -->
 
-- **Repository:** [https://github.com/gitter-lab/metl]
-- **Paper:** [https://www.biorxiv.org/content/10.1101/2024.03.15.585128v1]
-- **Demo:** [https://huggingface.co/spaces/gitter-lab/METL_demo]
+- **Repository:** [METL Repo](https://github.com/gitter-lab/metl)
+- **Paper:** [METL bio archrive](https://www.biorxiv.org/content/10.1101/2024.03.15.585128v1)
+- **Demo:** [Huggingface demo space for METL](https://huggingface.co/spaces/gitter-lab/METL_demo)
 
 ## How to Get Started with the Model
 
@@ -67,13 +65,7 @@ with torch.no_grad():
   
 ```
 
-## Training Details
-
-<!-- Do we want something here? -->
-
-## Citation [optional]
-
-<!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
+## Citation
 
 Biophysics-based protein language models for protein engineering
 Sam Gelman, Bryce Johnson, Chase Freschlin, Sameer D’Costa, Anthony Gitter, Philip A. Romero
@@ -81,3 +73,4 @@ bioRxiv 2024.03.15.585128; doi: https://doi.org/10.1101/2024.03.15.585128
 
 ## Model Card Contact
 
+For questions and comments about METL, the best way to reach out is through opening a github issue in the [METL repository](https://github.com/gitter-lab/metl/issues) issues page.