Update README.md

README.md

@@ -4,13 +4,16 @@ license: apache-2.0
 
 # TFv6 NavSim
 
-Autonomous driving planning model (TransFuser-based). Predicts waypoints and headings.
+Autonomous driving planning model (TransFuser-based). More information on [https://github.com/autonomousvision/lead](https://https://github.com/autonomousvision/lead).
 
 ## Usage
 
+Install dependencies
+
 ```bash
 pip install torch timm numpy opencv-python jaxtyping beartype omegaconf huggingface_hub
 ```
+
 ## Input/Output
 
 **Input:**
@@ -23,8 +26,6 @@ pip install torch timm numpy opencv-python jaxtyping beartype omegaconf huggingf
 - `waypoints`: (N, 2) predicted positions
 - `headings`: (N,) predicted angles
 
-## Details
+## Example
 
-- Architecture: TransFuser
-- Dataset: NavSim (4 cameras)
-- Checkpoint: Epoch 60
+See [example.ipynb](https://huggingface.co/ln2697/tfv6_navsim/blob/main/example.ipynb)