Update README.md

README.md

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