Upload README.md with huggingface_hub

README.md

@@ -24,7 +24,7 @@ tags:
   <br> <br>
 </div>
 
-This repository hosts Raster2Seq PyTorch checkpoints for polygon sequence generation from rasterized floorplans. Each checkpoint is stored in its own subfolder so users can download all checkpoints or only the subfolder they need.
+This repository hosts Raster2Seq's PyTorch checkpoints for floorplan reconstruction. Each checkpoint is stored in its own subfolder so users can download all checkpoints or only the subfolder they need.
 
 ## Available Checkpoints
 
@@ -60,7 +60,7 @@ local_repo = snapshot_download(
 With the Raster2Seq codebase, users can load by alias:
 
 ```bash
-python eval.py --checkpoint hf:raster2graph-512 ...
+python eval.py --checkpoint hf:cubicasa5k ...
 ```
 
 or download directly:
@@ -68,7 +68,7 @@ or download directly:
 ```python
 from raster2seq_hub import download_checkpoint
 
-ckpt_path = download_checkpoint("raster2graph-512")
+ckpt_path = download_checkpoint("cubicasa5k")
 ```
 
 **Please CITE** our paper and give us a :star: whenever this repository is used to help produce published results or incorporated into other software.