Update README.md

README.md

@@ -13,7 +13,7 @@ tags:
 
 # BindScreen Frozen
 
-This model corresponds to the BindScreen frozen configuration, in which both encoders are frozen and only the projection layers are trained on filtered ChEMBL.
+This model corresponds to the BindScreen frozen configuration, in which both encoders are frozen and only the projection layers are trained on filtered ChEMBL for LIT-PCBA.
 
 \[[Github Repo](https://github.com/pcdslab/BindScreen)\] | \[[Dataset on HuggingFace](https://huggingface.co/datasets/SaeedLab/BindScreen)\] | \[[Model Collection](https://huggingface.co/collections/SaeedLab/bindscreen)\] | \[[Cite](#citation)\]
 
@@ -91,7 +91,7 @@ with torch.no_grad():
 
 # bindscreen
 bindscreen = AutoModel.from_pretrained(
-  'SaeedLab/BindScreen-Frozen', 
+  'SaeedLab/BindScreen-Frozen-LIT_PCBA', 
   trust_remote_code=True
 ).eval()