Update README.md

README.md

@@ -26,10 +26,14 @@ model-index:
   ## Usage
 
   ```python
+  import gym
+  from huggingface_sb3 import load_from_hub 
   
-  model = load_from_hub(repo_id="dmenini/q-FrozenLake-v1-4x4-Slippery", filename="q-learning.pkl")
-
-  # Don't forget to check if you need to add additional attributes (is_slippery=False etc)
-  env = gym.make(model["env_id"])
+  model = load_from_hub(
+    repo_id="dmenini/q-FrozenLake-v1-4x4-Slippery",
+    filename="q-learning.pkl"
+  )
+  
+  env = gym.make("FrozenLake-v1", map_name="4x4", is_slippery=True)
   ```