ThomasSimonini HF staff commited on
Commit
82410fa
1 Parent(s): 669cf77

Remove max_steps as user configurable

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -34,7 +34,8 @@ def load_model(env_name):
34
 
35
  return model
36
 
37
- def replay(env_name, max_steps, time_sleep):
 
38
  env = load_env(env_name)
39
  model = load_model(env_name)
40
  #for i in range(num_episodes):
@@ -59,7 +60,7 @@ demo = gr.Interface(
59
  "SeaquestNoFrameskip-v4",
60
  "QbertNoFrameskip-v4",
61
  ]),
62
- gr.Slider(100, 10000, value=500),
63
  gr.Slider(0.01, 1, value=0.05),
64
  #gr.Slider(1, 20, value=5)
65
  ],
 
34
 
35
  return model
36
 
37
+ def replay(env_name, time_sleep):
38
+ max_steps = 500
39
  env = load_env(env_name)
40
  model = load_model(env_name)
41
  #for i in range(num_episodes):
 
60
  "SeaquestNoFrameskip-v4",
61
  "QbertNoFrameskip-v4",
62
  ]),
63
+ #gr.Slider(100, 10000, value=500),
64
  gr.Slider(0.01, 1, value=0.05),
65
  #gr.Slider(1, 20, value=5)
66
  ],