Update README.md

README.md

@@ -7,8 +7,18 @@ pipeline_tag: text-to-image
 # 🌌 GalaxySD: Can AI Dream of Unseen Galaxies? 
 # Conditional Diffusion Model for Galaxy Morphology Augmentation
 
-This is a text2galaxy tool, generating galaxy images given galaxy morphological prompts. 
+## Model Description
+
+This repository provides the trained weights for **GalaxySD**, a fine-tuned Stable Diffusion model designed to generate high-fidelity galaxy images from text prompts. The model demonstrates strong performance in synthesizing visually realistic galaxies that adhere closely to specified morphological characteristics, such as spiral arms, bulge prominence, and edge-on orientation.
+
+## Project Overview
+- 🏠 [Homepage](https://galaxysd-webpage.streamlit.app/): Detailed project introduction and interesting interactive quiz interface.
+- 📂 [GalaxySD Model Repository](https://github.com/chenruiRae/GalaxySD): Full model pipeline and training code.
+- 📊 [Training Dataset](https://zenodo.org/records/15669465): Preprocessed morphological galaxy data used for fine-tuning.
+
+## Tools and Contributions
+- 🛠️ Trained Galaxy Embedding Tool
+- 📚 Contributed Galaxy Morphology Catalog
+
 
-Visit the 🏠 [HOMEPAGE](https://galaxysd-webpage.streamlit.app/) to view the detailed introduction of our project.
 
-Our model pipeline 👉 [GalaxySD Repo](https://github.com/chenruiRae/GalaxySD).