Update README.md

README.md

@@ -25,12 +25,23 @@ 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 ...
-from huggingface_sb3 import load_from_hub
-
+model = A2C(policy = "MlpPolicy",
+            env = env,
+            gae_lambda = 0.9,
+            gamma = 0.99,
+            learning_rate = 0.00096,
+            max_grad_norm = 0.5,
+            n_steps = 8,
+            vf_coef = 0.4,
+            ent_coef = 0.0,
+            tensorboard_log = "./tensorboard",
+            policy_kwargs=dict(
+            log_std_init=-2, ortho_init=False),
+            normalize_advantage=False,
+            use_rms_prop= True,
+            use_sde= True,
+            verbose=1)
 ...
 ```