Update README.md

README.md

@@ -1,3 +1,24 @@
 ---
 license: mit
+language:
+- en
 ---
+# CycleGAN in Jax + Flax
+This repository contains weights for CycleGAN models that were trained on the horse2zebra and monet2photo TensorFlow datasets in Jax with Flax. 
+Please see the [Colab notebook](https://colab.research.google.com/github/louwjac/CycleGAN-Flax/blob/main/notebooks/inference.ipynb) for an example of how to instantiate the models and load the trained weights. 
+It includes a Gradio demo that makes it easy to submit external images to the model. You can also test the models in the [Spaces demo](https://huggingface.co/spaces/louwjac/CycleGan-Flax) without running any code yourself. 
+
+### TensorBoard logs
+The TensorBoard logs for the training runs can be viewed at the links below. The logged images are unfortunately not available due to file size constraints.
+1. [horse2zebra](https://tensorboard.dev/experiment/5tJZAfBLQkOJ66BUrBVBkQ/)
+2. [monet2photo](https://tensorboard.dev/experiment/GZZKiDoBTl6dsgsqRo7rZw/)
+
+
+<a href="https://huggingface.co/spaces/louwjac/CycleGan-Flax">
+  <img src="https://huggingface.co/datasets/huggingface/badges/raw/main/open-in-hf-spaces-sm-dark.svg" alt="Open in Spaces" style='display:inline;'>
+</a> 
+<a target="_blank" href="https://colab.research.google.com/github/louwjac/CycleGAN-Flax/blob/main/notebooks/inference.ipynb">
+  <img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"  style='display:inline;'/>
+</a>
+<a href="https://github.com/louwjac/CycleGAN-Flax">
+  <img alt="Custom badge" src="https://img.shields.io/static/v1?style=flat&logo=github&logoColor=white&labelColor=black&label=&message=View%20on%20GitHub&color=white" style='display:inline;'>