ThomasSimonini HF staff commited on
Commit
5d7151b
1 Parent(s): a1c192f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -25,7 +25,7 @@ def replay(model_id, filename, environment, evaluate):
25
  model= PPO.load(checkpoint, custom_objects=custom_objects)
26
 
27
  eval_env = make_atari_env(environment, n_envs=1)
28
- eval_env = VecFrameStack(env, n_stack=4)
29
 
30
  video_folder = 'logs/videos/'
31
  video_length = 100
 
25
  model= PPO.load(checkpoint, custom_objects=custom_objects)
26
 
27
  eval_env = make_atari_env(environment, n_envs=1)
28
+ eval_env = VecFrameStack(eval_env, n_stack=4)
29
 
30
  video_folder = 'logs/videos/'
31
  video_length = 100