Spaces:
Runtime error
Runtime error
Thomas Simonini
commited on
Commit
·
5871174
1
Parent(s):
d812b5e
Update app.py
Browse files
app.py
CHANGED
|
@@ -39,7 +39,7 @@ def replay(model_id, filename, environment, evaluate):
|
|
| 39 |
env.reset()
|
| 40 |
for _ in range(video_length + 1):
|
| 41 |
action, _states = model.predict(obs)
|
| 42 |
-
|
| 43 |
# Save the video
|
| 44 |
env.close()
|
| 45 |
|
|
|
|
| 39 |
env.reset()
|
| 40 |
for _ in range(video_length + 1):
|
| 41 |
action, _states = model.predict(obs)
|
| 42 |
+
obs, rewards, dones, info = env.step(action)
|
| 43 |
# Save the video
|
| 44 |
env.close()
|
| 45 |
|