Update README.md

README.md

@@ -14,13 +14,13 @@ widget:
 
 # DreamBooth model for Starcraft:Remastered terrain
 
-This is a Stable Diffusion model fine-tuned on Starcraft terrain images on the Space Platform tileset with DreamBooth. It can be used by modifying the `instance_prompt`: **isometric scspace terrain**
+This is a Stable Diffusion model fine-tuned on Starcraft terrain images on the Space Platform tileset with DreamBooth. It can be used by adding the `instance_prompt`: **isometric scspace terrain**
 
-It was trained on 32x32 terrain images from 265 melee maps including original Blizzard maps and those downloaded from scmscx.com and broodwarmaps.com.
+It was trained on 32x32 terrain images from 265 melee maps including original Blizzard maps and those downloaded from scmscx.com and broodwarmaps.net.
 
 To run the demo with the ability to generate map files directly/with more coherence, use this notebook on Colab:
 
-<ADD_NOTEBOOK_LINK_HERE>
+[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/wdcqc/WaveFunctionDiffusion/blob/remaster/WaveFunctionDiffusion_Demo.ipynb)
 
 In addition to Dreambooth, a custom VAE model (`AutoencoderTile`) is trained to encode and decode the latents to/from tileset probabilities ("waves") and then generated as Starcraft maps.