Update README.md

README.md

@@ -20,9 +20,7 @@ library_name: transformers
 
 This repository provides PyTorch source code associated with our publication, "A Mamba-Based Foundation Model for Chemistry".
 
-**Paper:** [Arxiv Link](https://openreview.net/pdf?id=HTgCs0KSTl)
-
-**HuggingFace:** [HuggingFace Link](https://huggingface.co/ibm/materials.smi_ssed)
+**Paper NeurIPS AI4Mat 2024:** [Arxiv Link](https://openreview.net/pdf?id=HTgCs0KSTl)
 
 For more information contact: [email protected] or [email protected].
 
@@ -54,7 +52,7 @@ For more information contact: [email protected] or [email protected].
 
 ### Pretrained Models and Training Logs
 
-We provide checkpoints of the SMI-SSED model pre-trained on a dataset of ~91M molecules curated from PubChem. The pre-trained model shows competitive performance on classification and regression benchmarks from MoleculeNet. For model weights: [HuggingFace Link](https://huggingface.co/ibm/materials.smi-ted)
+We provide checkpoints of the SMI-SSED model pre-trained on a dataset of ~91M molecules curated from PubChem. The pre-trained model shows competitive performance on classification and regression benchmarks from MoleculeNet.
 
 Add the SMI-SSED `pre-trained weights.pt` to the `inference/` or `finetune/` directory according to your needs. The directory structure should look like the following: