Spaces:
Runtime error
Runtime error
Commit
•
4a80994
1
Parent(s):
a53beac
Update app.py
Browse files
app.py
CHANGED
@@ -36,7 +36,7 @@ def replay(model_id, filename, environment, evaluate):
|
|
36 |
record_video_trigger=lambda x: x == 0, video_length=video_length,
|
37 |
name_prefix=f"test")
|
38 |
|
39 |
-
env.reset()
|
40 |
for _ in range(video_length + 1):
|
41 |
action, _states = model.predict(obs)
|
42 |
obs, rewards, dones, info = env.step(action)
|
|
|
36 |
record_video_trigger=lambda x: x == 0, video_length=video_length,
|
37 |
name_prefix=f"test")
|
38 |
|
39 |
+
obs = env.reset()
|
40 |
for _ in range(video_length + 1):
|
41 |
action, _states = model.predict(obs)
|
42 |
obs, rewards, dones, info = env.step(action)
|