Update README.md

README.md

@@ -5,7 +5,7 @@ license: mit
 
 A PyTorch-based Generative Adversarial Network (GAN) for training and generating pixel art images.
 
-## Setup
+## Setup for model training
 
 Git clone the pixel repo:
 ```
@@ -17,11 +17,15 @@ Get inside the cloned folder:
 cd pixel
 ```
 
+## Train your model
+
 Start training with your dataset (in ./data/):
 ```
 python trainer.py
 ```
 
+## Interact with the post-trained model
+
 When finished, check the model file (in ./models/) then run the inference:
 ```
 python generator.py