Update README.md

README.md

@@ -12,7 +12,7 @@ license: mit
 - **License:** mit
 
 
-### Model Sources
+### Source Code
 
 <!-- Provide the basic links for the model. -->
 
@@ -20,19 +20,35 @@ license: mit
 
 ### Assets List
 
-- `mimic_iv_text_embed.csv`: Pre saved embedding table for mimic text report.
-- `mimic_vae.pt` & `mimic_vae_lite.pt`: Dictionary consists of VAE encoding latents of mimic iv ecg dataset. `lite` version contains only 50 000 samples for convenience.
-- `unet.pth`: A trained noise predictor model checkpoints. Use both VAE latent and patient characteristc info.
-- `vae_model.pth`: A trained VAE model checkpoints suit for the unet model and latent datasets above.
-- `sample_config.json`: An example of configuration.
+- **prerequisites_for_inference.zip**:
+```
+prerequisites
+├── clf_data : Directory contains data for imbalanced classification experiments 
+├── clip_model.pth : CLIP model checkpoints.
+├── mimic_vae_lite_0_new.pt : Subsets of VAE encoding latents of MIMIC IV ECG dataset.
+├── ptbxl_text_embed.csv : Pre-generated text embeddings of PTB-XL dataset
+├── ptbxl_vae.pt : VAE encoding latents of PTB-XL dataset.
+├── unet_all.pth : A pretrained full capacity DiffuSETS model.
+└── vae_model.pth : VAE model checkpoints suit for the unet model and latent datasets above.
+```
+
+- **mimic_vae_lite_0_new.pt**: VAE encoding latents of MIMIC IV ECG dataset. Only needed while training.
+
+## Usage
 
 
-## Quick Uses
+**Quick Start:** Download and unzip the prerequisites_for_inference.zip, place the `./prerequisites` folder under repo's root, then start ECG generation through
 
-Download the all the prerequisites into the `./prerequisites` folder, then start ECG generating through
+```sh
+python DiffuSETS_inference.py path/to/your/config.json
+```
+
+or
 
 ```sh
-python DiffuSETS_inference.py ./prerequisites/sample_config.json
+python -m test_scripts.diversity
 ```
 
-For more details in training and inference, please refer to our [github repo page](https://github.com/Raiiyf/text2ecg)
+**For users want to train from scratch:** Besides above prerequisites_for_inference.zip, the mimic_vae_0_new.pt (training data encoded by vae) should also be downloaded and put into the `./prerequisites` folder.
+
+For more details about training and inference, please refer to our [github repo page](https://github.com/Raiiyf/text2ecg)