Update README.md

README.md

@@ -9,23 +9,24 @@ This is a trained model of a PPO agent playing PongNoFrameskip-v4 using the stab
 
 <video src="https://huggingface.co/ThomasSimonini/ppo-PongNoFrameskip-v4/resolve/main/output.mp4" controls autoplay loop></video>
 
-# Usage (with Stable-baselines3)
-
-
 ## Evaluation Results
 Mean_reward = 21.00 +/- 0.0
 
-## Watch your agent interacts
+# Usage (with Stable-baselines3)
+## Watch your agent interacts (in Google Colab)
 - You need to use `gym==0.19` since it **includes Atari Roms**.
 - The Actor Space is 6 since we use only **legit actions**.
 
 ```python
-# Install these libraries (don't forget to restart the runtime after installing the librairies)
+# Install these libraries in one cell (don't forget to restart the runtime after installing the librairies)
 !pip install stable-baselines3[extra]
 !pip install huggingface_sb3
 !pip install huggingface_hub
 !pip install pickle5
+```
 
+Don't forget to restart the runtime before running the code below:
+```python
 # Import the libraries
 import os