Update README.md

README.md

@@ -35,7 +35,8 @@ This project is licensed under the MIT License.
 The GAN consists of two parts: a generator and a discriminator. Both models were trained on the Aligned Bach Chorales Dataset, which represents Bach chorales in a binary matrix format.
 
 ## Installation and Requirements
-To use this model, you need to have PyTorch version 2.1.0+cu121 installed. You can install PyTorch by following the instructions on the [official website](https://pytorch.org/get-started/locally/).
+To use this model, you need to have PyTorch version 2.1.0+cu121 installed. The dataset loader, model structure, and training scripts are all included in the torch_binary2_v1 notebook.
+
 
 ## Usage
 1. **Clone the Repository**: 
@@ -43,7 +44,7 @@ To use this model, you need to have PyTorch version 2.1.0+cu121 installed. You c
 git clone [repository URL]
 ```
 2. **Load the Models**: 
-- Use `torch.load()` to load the provided state dictionaries for both the generator and the discriminator.
+- Use torch_binary2_v1 notebook to load the provided state dictionaries for both the generator and the discriminator.
 3. **Convert Binary Matrices to MIDI**: 
 - Utilize the `binary_to_midi` notebook included in this repository to convert the binary matrices generated by the GAN into MIDI format.