Update README.md

README.md

@@ -11,18 +11,13 @@ license: mit
 
 # Model Card for LegoGPT
 
-<!-- Provide a quick summary of what the model is/does. -->
-
-
+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).
+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
 
 ### Model Description
 
-<!-- Provide a longer summary of what this model is. -->
-
-
-
 - **Developed by:** [Carnegie Mellon University Generative Intelligence Lab](https://www.cs.cmu.edu/~generative-intelligence-lab/)
 - **Funded by:** This work is partly supported by the Packard Foundation, Cisco Research Grant, and Amazon Faculty Award. This work is also in part supported by the Manufacturing Futures Institute, Carnegie Mellon University, through a grant from the Richard King Mellon Foundation. KD is supported by the Microsoft Research PhD Fellowship.
 - **Model type:** Autoregressive
@@ -32,45 +27,17 @@ license: mit
 
 ### Model Sources
 
-<!-- Provide the basic links for the model. -->
-
 - **Repository:** [AvaLovelace1/LegoGPT](https://github.com/AvaLovelace1/LegoGPT)
 - **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)
 
-## Uses
-
-<!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
-
-### Direct Use
-
-<!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
-
-[More Information Needed]
+## Limitations
 
-### Out-of-Scope Use
-
-<!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
-
-[More Information Needed]
-
-## Bias, Risks, and Limitations
-
-<!-- This section is meant to convey both technical and sociotechnical limitations. -->
-
-[More Information Needed]
-
-### Recommendations
-
-<!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
-
-Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
+The model is restricted to creating structures on a 20x20x20 grid. It was trained on a dataset of 21 object categories: *basket, bed, bench, birdhouse, bookshelf, bottle, bowl, bus, camera, car, chair, guitar, jar, mug, piano, pot, sofa, table, tower, train, vessel*. Performance on prompts from outside these categories may be limited.
 
 ## How to Get Started with the Model
 
-Use the code below to get started with the model.
-
-[More Information Needed]
+See the [GitHub repo](https://github.com/AvaLovelace1/LegoGPT) for usage examples and an interactive CLI demo.
 
 ## Training Details