Update README.md

README.md

@@ -1,3 +1,38 @@
----
-license: mit
----
+---
+license: mit
+---
+
+# Model Card for DiffuSETS
+
+## Description
+
+<!-- Provide a longer summary of what this model is. -->
+
+- **Developed by:** Yongfan Lai
+- **License:** mit
+
+
+### Model Sources
+
+<!-- Provide the basic links for the model. -->
+
+- **Repository:** [Github](https://github.com/Raiiyf/text2ecg)
+
+### 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.
+
+
+## Quick Uses
+
+Download the all the prerequisites into the `./prerequisites` folder, then start ECG generating through
+
+```sh
+python DiffuSETS_inference.py ./prerequisites/sample_config.json
+```
+
+For more details in training and inference, please refer to our [github repo page](https://github.com/Raiiyf/text2ecg)