Update README.md

README.md

@@ -7,6 +7,10 @@ tags:
 - pytorch_model_hub_mixin
 ---
 
+# GSFM
+
+Trained on millions of gene sets automatically extracted from literature and raw RNA-seq data, GSFM learns to recover held-out genes from gene sets. The resulting model exhibits state of the art performance on gene function prediction.
+
 ## Website
 
 <https://gsfm.maayanlab.cloud/>
@@ -14,6 +18,7 @@ tags:
 ## Usage
 
 ```bash
+# install gsfm python library from its source on huggingface
 GIT_LFS_SKIP_SMUDGE=1 pip install git+https://huggingface.co/maayanlab/gsfm
 ```
 
@@ -21,6 +26,7 @@ GIT_LFS_SKIP_SMUDGE=1 pip install git+https://huggingface.co/maayanlab/gsfm
 import torch
 from gsfm import Vocab, GSFM
 
+# load gsfm vocabulary and model weights
 vocab = Vocab.from_pretrained('maayanlab/gsfm')
 model = GSFM.from_pretrained('maayanlab/gsfm')