Update README.md

README.md

@@ -1,7 +1,7 @@
 ---
 base_model: meta-llama/Llama-3.2-1B-Instruct
 datasets:
-- AvaLovelace/StableText2Lego
+- AvaLovelace/StableText2Brick
 language:
 - en
 library_name: peft
@@ -9,9 +9,9 @@ license: mit
 pipeline_tag: text-to-3d
 ---
 
-# Model Card for LegoGPT
+# Model Card for BrickGPT
 
-These are the model weights for LegoGPT, the first approach for generating physically stable LEGO brick models from text prompts, as described in [Generating Physically Stable and Buildable LEGO® Designs from Text](https://huggingface.co/papers/2505.05469).
+These are the model weights for BrickGPT, the first approach for generating physically stable toy brick models from text prompts, as described in [Generating Physically Stable and Buildable LEGO® Designs from Text](https://huggingface.co/papers/2505.05469).
 This model was fine-tuned from [meta-llama/Llama-3.2-1B-Instruct](https://huggingface.co/meta-llama/Llama-3.2-1B-Instruct).
 
 ## Model Details
@@ -24,13 +24,13 @@ This model was fine-tuned from [meta-llama/Llama-3.2-1B-Instruct](https://huggin
 - **Language(s):** English
 - **License:** MIT
 - **Finetuned from model:** [meta-llama/Llama-3.2-1B-Instruct](https://huggingface.co/meta-llama/Llama-3.2-1B-Instruct)
-- **Project page:** https://avalovelace1.github.io/LegoGPT/
+- **Project page:** https://avalovelace1.github.io/BrickGPT/
 
 ### Model Sources
 
-- **Repository:** [AvaLovelace1/LegoGPT](https://github.com/AvaLovelace1/LegoGPT)
+- **Repository:** [AvaLovelace1/BrickGPT](https://github.com/AvaLovelace1/BrickGPT)
 - **Paper:** [Generating Physically Stable and Buildable LEGO® Designs from Text](https://huggingface.co/papers/2505.05469)
-- **Demo:** [cmu-gil/LegoGPT-Demo](https://huggingface.co/spaces/cmu-gil/LegoGPT-Demo)
+- **Demo:** [cmu-gil/BrickGPT-Demo](https://huggingface.co/spaces/cmu-gil/BrickGPT-Demo)
 
 ## Limitations
 
@@ -38,13 +38,13 @@ The model is restricted to creating structures made of 1-unit-tall cuboid bricks
 
 ## How to Get Started with the Model
 
-See the [GitHub repo](https://github.com/AvaLovelace1/LegoGPT) for usage examples and an interactive CLI demo.
+See the [GitHub repo](https://github.com/AvaLovelace1/BrickGPT) for usage examples and an interactive CLI demo.
 
 ## Training Details
 
 ### Training Data
 
-LegoGPT was trained using [StableText2Lego](https://huggingface.co/datasets/AvaLovelace/StableText2Lego), a dataset of 47k LEGO structures.
+BrickGPT was trained using [StableText2Brick](https://huggingface.co/datasets/AvaLovelace/StableText2Brick), a dataset of 47k toy brick structures.
 
 ### Training Procedure
 
@@ -75,7 +75,7 @@ See the [paper](https://huggingface.co/papers/2505.05469) for detailed evaluatio
 If you find this model useful for your research, please cite the following work.
 
 ```bibtex
-@article{pun2025legogpt,
+@article{pun2025brickgpt,
     title   = {Generating Physically Stable and Buildable LEGO Designs from Text},
     author  = {Pun, Ava and Deng, Kangle and Liu, Ruixuan and Ramanan, Deva and Liu, Changliu and Zhu, Jun-Yan},
     journal = {arXiv preprint arXiv:2505.05469},