Update README.md

README.md

@@ -7,7 +7,31 @@ tags:
 - pytorch_model_hub_mixin
 ---
 
-This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
-- Code: [More Information Needed]
-- Paper: [More Information Needed]
-- Docs: [More Information Needed]
+## Website
+
+<https://gsfm.maayanlab.cloud/>
+
+## Usage
+
+```python
+import torch
+
+vocab = Vocab.from_pretrained('maayanlab/gsfm')
+model = GSFM.from_pretrained('maayanlab/gsfm')
+
+# convert gene symbols into token ids
+token_ids = torch.tensor(vocab(['ACE1', 'ACE2']))[None, :]
+
+# use model to predict missing genes from the set
+logits = torch.squeeze(model(token_ids))
+top_10 = sorted(zip(logits, vocab.vocab))[-10:]
+top_10
+
+# get gene embedding
+gene_embeddings = model.embedding(token_ids)
+gene_embeddings
+
+# get model middle layer
+gene_set_encoding = model.encode(token_ids)
+gene_set_encoding
+```