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

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -31,7 +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="ppo-SpaceInvadersNoFrameskip-v4")
35
  model = PPO.load(checkpoint)
36
  ```
37
 
 
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.zip")
35
  model = PPO.load(checkpoint)
36
  ```
37