Views
No views yet
1import gymnasium as gym
2
3from stable_baselines3 import PPO
4from huggingface_sb3 import load_from_hub
5
6model_filename = load_from_hub(repo_id, 'models/ppo-LunarLander-v3/best_model.zip')
7
8model = PPO.load(model_filename)
9
10env = gym.make(**LunarLander-v3**, render_mode="human")
11 obs, info = env.reset()
12
13 done = False
14
15 while not done:
16 action, _states = model.predict(obs, deterministic=True)
17 obs, rewards, done, truncated, info = env.step(action)
18 env.render()
19
20 env.close()