Minor changes (#11)

- Update README.md (7bdc4b7d2d6d7e45f9a0d50e8865115bdcea7697)
- Update test_huggingface_mae.py (0b368c3c595e5b9b1d7939b3ae5b0aa22e34bfb6)

README.md

@@ -21,7 +21,7 @@ This model is a [channel-agnostic masked autoencoder](https://openaccess.thecvf.
 - **Developed, funded, and shared by:** Recursion
 - **Model type:** Vision transformer CA-MAE
 - **Image modality:** Optimized for microscopy images from the CellPainting assay
-- **License:** Attribution-NonCommercial 4.0 International
+- **License:** [Non-Commercial End User License Agreement](https://huggingface.co/recursionpharma/OpenPhenom/blob/main/LICENSE)
 
 
 ### Model Sources
@@ -67,7 +67,7 @@ import torch
 
 from huggingface_mae import MAEModel
 
-huggingface_phenombeta_model_dir = "."
+huggingface_openphenom_model_dir = "."
 # huggingface_modelpath = "recursionpharma/OpenPhenom"
 
 
@@ -75,7 +75,7 @@ huggingface_phenombeta_model_dir = "."
 def huggingface_model():
     # Make sure you have the model/config downloaded from https://huggingface.co/recursionpharma/OpenPhenom to this directory
     # huggingface-cli download recursionpharma/OpenPhenom --local-dir=.
-    huggingface_model = MAEModel.from_pretrained(huggingface_phenombeta_model_dir)
+    huggingface_model = MAEModel.from_pretrained(huggingface_openphenom_model_dir)
     huggingface_model.eval()
     return huggingface_model
 
@@ -96,6 +96,8 @@ def test_model_predict(huggingface_model, C, return_channelwise_embeddings):
     assert embeddings.shape == (2, expected_output_dim)
 ```
 
+**Note: Currently, the model cannot be loaded via the `AutoModel` available in HuggingFace transformers library**
+
 
 ## Training, evaluation and testing details
 

test_huggingface_mae.py

@@ -3,7 +3,7 @@ import torch
 
 from huggingface_mae import MAEModel
 
-huggingface_phenombeta_model_dir = "."
+huggingface_openphenom_model_dir = "."
 # huggingface_modelpath = "recursionpharma/test-pb-model"
 
 
@@ -11,7 +11,7 @@ huggingface_phenombeta_model_dir = "."
 def huggingface_model():
     # Make sure you have the model/config downloaded from https://huggingface.co/recursionpharma/test-pb-model to this directory
     # huggingface-cli download recursionpharma/test-pb-model --local-dir=.
-    huggingface_model = MAEModel.from_pretrained(huggingface_phenombeta_model_dir)
+    huggingface_model = MAEModel.from_pretrained(huggingface_openphenom_model_dir)
     huggingface_model.eval()
     return huggingface_model