diff --git a/README.md b/README.md
index 8ade768..f721ab0 100644
--- a/README.md
+++ b/README.md
@@ -1,6 +1,40 @@
-# enhancer
-Enhancer is a Pytorch-based opensource toolkit for speech enhancement. It is designed to save time for audio researchers. Is provides easy to use pretrained audio enhancement models and facilitates highly customisable custom model training . Enhancer provides
+# mayavoz
+mayavoz is a Pytorch-based opensource toolkit for speech enhancement. It is designed to save time for audio researchers. Is provides easy to use pretrained audio enhancement models and facilitates highly customisable custom model training .
 
-* Various pretrained models nicely integrated with huggingface that users can select and use without any hastle.
-* Ability to train and validation your own custom speech enhancement models with just under 10 lines of code!
-* A command line tool that facilitates training of highly customisable speech enhacement models from the terminal itself!
+| **[Quick Start]()** | **[Installation]()** | **[Tutorials]()** | **[Available Recipes]()**
+## Key features :key:
+
+* Various pretrained models nicely integrated with huggingface 	:hugs: that users can select and use without any hastle.
+* :package: Ability to train and validation your own custom speech enhancement models with just under 10 lines of code!
+* :magic_wand: A command line tool that facilitates training of highly customisable speech enhacement models from the terminal itself!
+* :zap: Supports multi-gpu training integrated with Pytorch Lightning.
+
+## Quick Start :fire:
+``` python
+from mayavoz import Mayamodel
+
+model = Mayamodel.from_pretrained("mayavoz/waveunet")
+model("noisy_audio.wav")
+```
+
+## Installation
+Only Python 3.8+ is officially supported (though it might work with Python 3.7)
+
+- With Pypi
+```
+pip install mayavoz
+```
+
+- With conda
+
+```
+conda env create -f environment.yml
+conda activate mayavoz
+```
+
+- From source code
+```
+git clone url
+cd mayavoz
+pip install -e .
+```
diff --git a/enhancer/models/model.py b/enhancer/models/model.py
index 4f1d75b..7dbc065 100644
--- a/enhancer/models/model.py
+++ b/enhancer/models/model.py
@@ -110,6 +110,7 @@ class Model(pl.LightningModule):
 
     def setup(self, stage: Optional[str] = None):
         if stage == "fit":
+            torch.cuda.empty_cache()
             self.dataset.setup(stage)
             self.dataset.model = self