Update app.py
Browse files
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,
|
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)
|