ThomasSimonini HF staff commited on
Commit
e61d111
·
1 Parent(s): 3055a57

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -2
README.md CHANGED
@@ -31,8 +31,7 @@ from stable_baselines3.common.vec_env import VecFrameStack
31
  # Retrieve the model from the hub
32
  ## repo_id = id of the model repository from the Hugging Face Hub (repo_id = {organization}/{repo_name})
33
  ## filename = name of the model zip file from the repository
34
- checkpoint = load_from_hub(repo_id="ThomasSimonini/ppo-SpaceInvadersNoFrameskip-v4", filename="SpaceInvadersNoFrameskip-v4")
35
- print(checkpoint)
36
  model = PPO.load(checkpoint)
37
  ```
38
 
 
31
  # Retrieve the model from the hub
32
  ## repo_id = id of the model repository from the Hugging Face Hub (repo_id = {organization}/{repo_name})
33
  ## filename = name of the model zip file from the repository
34
+ checkpoint = load_from_hub(repo_id="ThomasSimonini/ppo-SpaceInvadersNoFrameskip-v4", filename="ppo-SpaceInvadersNoFrameskip-v4")
 
35
  model = PPO.load(checkpoint)
36
  ```
37