Update README.md

README.md

@@ -26,12 +26,25 @@ This is a trained model of a **A2C** agent playing **AntBulletEnv-v0**
 using the [stable-baselines3 library](https://github.com/DLR-RM/stable-baselines3).
 
 ## Usage (with Stable-baselines3)
-TODO: Add your code
 
 
 ```python
-from stable_baselines3 import ...
+import gym
+
+from stable_baselines3 import A2C
 from huggingface_sb3 import load_from_hub
 
-...
+checkpoint = load_from_hub(
+    repo_id="AntBulletEnv-v0",
+    filename="a2c-AntBulletEnv-v0.zip",
+)
+model = A2C.load(checkpoint)
+
+# Evaluate the agent and watch it
+eval_env = gym.make("AntBulletEnv-v0")
+mean_reward, std_reward = evaluate_policy(
+    model, eval_env, render=True, n_eval_episodes=5, deterministic=True, warn=False
+)
+print(f"mean_reward={mean_reward:.2f} +/- {std_reward}")
+
 ```