riiswa commited on
Commit
f624fe7
·
verified ·
1 Parent(s): afdd3a1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -54,7 +54,7 @@ if __name__ == "__main__":
54
  if env_name == "Swimmer-v3" or env_name == "Walker2d-v3":
55
  agent = "trpo"
56
 
57
- dataset_path, video_path = generate_dataset_from_expert(agent, _env_name, 50, 3)
58
  return video_path, gr.Button("Compute the symbolic policy!", interactive=True), {
59
  "dataset_path": dataset_path,
60
  "ipe": None,
@@ -147,4 +147,4 @@ if __name__ == "__main__":
147
  symbolic_policy, inputs=[state], outputs=[sym_video, sym_formula, button]
148
  )
149
  app.queue(default_concurrency_limit=3, max_size=10)
150
- app.launch()
 
54
  if env_name == "Swimmer-v3" or env_name == "Walker2d-v3":
55
  agent = "trpo"
56
 
57
+ dataset_path, video_path = generate_dataset_from_expert(agent, _env_name, 25, 3)
58
  return video_path, gr.Button("Compute the symbolic policy!", interactive=True), {
59
  "dataset_path": dataset_path,
60
  "ipe": None,
 
147
  symbolic_policy, inputs=[state], outputs=[sym_video, sym_formula, button]
148
  )
149
  app.queue(default_concurrency_limit=3, max_size=10)
150
+ app.launch(show_error=True)