Update README.md

README.md

@@ -10,14 +10,20 @@ tags:
 # Musika Model: halvany\_oszi\_rozsa
 ## Model provided by: DarkDude31
 
-Finetuned (from misc) halvany\_oszi\_rozsa model for the [Musika system](https://github.com/marcoppasini/musika) for fast infinite waveform music generation.
+Fine-tuned (from misc) halvany\_oszi\_rozsa model for the [Musika system](https://github.com/marcoppasini/musika) for fast infinite waveform music generation.
 Introduced in [this paper](https://arxiv.org/abs/2208.08706).
 
 ## How to use
 
-You can generate music from this finetuned (from misc) musika_halvany_oszi_rozsa model using the notebook available [here](https://colab.research.google.com/drive/1HJWliBXPi-Xlx3gY8cjFI5-xaZgrTD7r).
+You can generate music from this fine-tuned (from misc) halvany_oszi_rozsa model using the notebook available [here](https://colab.research.google.com/drive/1HJWliBXPi-Xlx3gY8cjFI5-xaZgrTD7r).
+
+Only the `gen_ema.h5` file is needed to generate music. Place it in your `checkpoints` folder.
 
 ### Model description
 
 This pretrained GAN system consists of a ResNet-style generator and discriminator. During training, stability is controlled by adapting the strength of gradient penalty regularization on-the-fly. The gradient penalty weighting term is contained in *switch.npy*. The generator is conditioned on a latent coordinate system to produce samples of arbitrary length. The latent representations produced by the generator are then passed to a decoder which converts them into waveform audio.
 The generator has a context window of about 12 seconds of audio.
+
+### Training description
+
+I had trained this on colab for 5 epochs. (about 5 \* 9000 iterations)