Update README.md

README.md

@@ -8,14 +8,14 @@ pipeline_tag: robotics
 
 VQ-BeT (as per [Behavior Generation with Latent Actions](https://arxiv.org/abs/2403.03181)) trained for the `PushT` environment from [gym-pusht](https://github.com/huggingface/gym-pusht).
 
-![demo](demo.gif)
-
 ## How to Get Started with the Model
 
 See the [LeRobot library](https://github.com/huggingface/lerobot) (particularly the [evaluation script](https://github.com/huggingface/lerobot/blob/main/lerobot/scripts/eval.py)) for instructions on how to load and evaluate this model.
 
 ## Training Details
 
+Trained with [LeRobot@342f429](https://github.com/huggingface/lerobot/tree/342f429f1c321a2b4501c3007b1dacba7244b469).
+
 The model was trained using this command:
 
 ```bash
@@ -40,7 +40,9 @@ The model was evaluated on the `PushT` environment from [gym-pusht](https://gith
 - Maximum overlap with target (seen as `eval/avg_max_reward` in the charts above). This ranges in [0, 1].
 - Success: whether or not the maximum overlap is at least 95%.
 
+The results of each of the individual rollouts may be found in [eval_info.json](eval_info.json).
+
 <blank>|Ours
 -|-
-Average max. overlap ratio | 0.887
+Average max. overlap ratio for 500 episodes | 0.887
 Success rate for 500 episodes (%) | 66.0