Update README.md
Browse files
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.
|