Update README.md

README.md

@@ -25,16 +25,13 @@ model-index:
 This is a trained model of a **PPO** agent playing **LunarLander-v2**
 using the [stable-baselines3 library](https://github.com/DLR-RM/stable-baselines3).
 
-## Usage (with Stable-baselines3)
-TODO: Add your code
-
-
 ```python
+!pip install shimmy
 from stable_baselines3 import ...
 from huggingface_sb3 import load_from_hub
 
-repo_id = "JuanMa360/ppo-LunarLander-v2" # The repo_id
-filename = "ppo-LunarLander-v2.zip" # The model filename.zip
+repo_id = "JuanMa360/ppo-LunarLander-v2"
+filename = "ppo-LunarLander-v2.zip" 
 
 # When the model was trained on Python 3.8 the pickle protocol is 5
 # But Python 3.6, 3.7 use protocol 4