Update README.md

README.md

@@ -18,7 +18,7 @@ inference. Model input is a SMILES string of the small molecule, and the output
 each of the toxic effects of the Tox21 dataset. 
 
 
-**Important:** For leaderboard submission, your Space needs to include training code. The file `train.py` should train the model using the config specified inside the `config/` folder and save the final model parameters into a file inside the `checkpoints/` folder. The model should be trained using the [Tox21_dataset](https://huggingface.co/datasets/tschouis/tox21) provided on Hugging Face. The datasets can be loaded like this:
+**Important:** For leaderboard submission, your Space needs to include training code. The file `train.py` should train the model using the config specified inside the `config/` folder and save the final model parameters into a file inside the `checkpoints/` folder. The model should be trained using the [Tox21_dataset](https://huggingface.co/datasets/ml-jku/tox21) provided on Hugging Face. The datasets can be loaded like this:
 ```python
 from datasets import load_dataset
 ds = load_dataset("ml-jku/tox21", token=token)