Spaces:
Sleeping
Sleeping
File size: 871 Bytes
0f97cc0 c785ba7 0f97cc0 c785ba7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
import streamlit as st
import gymnasium
from gymnasium.wrappers.record_video import RecordVideo
from PIL import Image
import time
steps = st.number_input("Step Number", min_value=1, value=100)
sim_clicked = st.button("Simulate")
if sim_clicked:
env = gymnasium.make("MountainCarContinuous-v0", render_mode="rgb_array")
env = RecordVideo(env, video_folder="./videos")
obs, info = env.reset()
with st.empty():
for i in range(steps):
action = env.action_space.sample()
obs, reward, terminated, truncated, info = env.step(action)
rgb_array = env.render()
image = Image.fromarray(rgb_array)
st.image(image, caption=f"Step {i}")
time.sleep(0.02)
if terminated or truncated:
break
env.close()
st.video("./videos/rl-video-episode-0.mp4")
|