Update README.md

README.md

@@ -22,19 +22,19 @@ model-index:
 ---
 # **Q-Learning** Agent playing **CartPole-v1**
 
-This is a trained model of a **Q-Learning** agent playing **CartPole-v1** .
+This is a trained model of a **Reinforce** agent playing **CartPole-v1** .
 
 ## Usage
 
 ```python
 
-model = load_from_hub(repo_id="sayby/Reinforce-CartPole-v1", filename="q-learning.pkl")
+model = load_from_hub(repo_id="sayby/Reinforce-CartPole-v1", filename="model.pt")
 
 # Don't forget to check if you need to add additional attributes (is_slippery=False etc)
 
 env = gym.make(model["env_id"])
 
-evaluate_agent(env, model["max_steps"], model["n_eval_episodes"], model["qtable"], model["eval_seed"])
+evaluate_agent(env, model["max_steps"], model["n_eval_episodes"], model["eval_seed"])
 
  
 ```