Emaad commited on
Commit
76efb5f
1 Parent(s): 844f730

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +7 -0
README.md CHANGED
@@ -25,6 +25,13 @@ We trained on the [Human Protein Atlas](https://www.proteinatlas.org) (HPA) and
25
  CELL-E 2 utilizes pretrained amino acid embeddings from [ESM-2](https://github.com/facebookresearch/esm).
26
  Localization is predicted as a binary image atop the provided nucleus. The logit values are weighted against these binary images to produce a heatmap of expected localization.
27
 
 
 
 
 
 
 
 
28
  ## Model variations
29
 
30
  We have made several versions of CELL-E 2 available. The naming scheme follows the structure ```training set_hidden size``` where the hidden size is set to the embedding dimension of the pretrained ESM-2 model.
 
25
  CELL-E 2 utilizes pretrained amino acid embeddings from [ESM-2](https://github.com/facebookresearch/esm).
26
  Localization is predicted as a binary image atop the provided nucleus. The logit values are weighted against these binary images to produce a heatmap of expected localization.
27
 
28
+ ## Spaces
29
+
30
+ We have two spaces available where you can run predictions on your own data!
31
+
32
+ - [Image Prediction](https://huggingface.co/spaces/HuangLab/CELL-E_2-Image_Prediction)
33
+ - [Sequence Prediction](https://huggingface.co/spaces/HuangLab/CELL-E_2-Sequence_Prediction)
34
+
35
  ## Model variations
36
 
37
  We have made several versions of CELL-E 2 available. The naming scheme follows the structure ```training set_hidden size``` where the hidden size is set to the embedding dimension of the pretrained ESM-2 model.