Spaces:
Runtime error
Runtime error
| import gymnasium as gym | |
| import streamlit as st | |
| # import cv2 | |
| st.write(st.__version__ + "omg") | |
| # env = gym.make("CartPole-v1", render_mode='rgb_array') | |
| # observation, info = env.reset(seed=42) | |
| # rgb_array = [] | |
| # for _ in range(1000): | |
| # action = env.action_space.sample() | |
| # observation, reward, terminated, truncated, info = env.step(action) | |
| # rgb_array.append(env.render()) | |
| # if terminated or truncated: | |
| # observation, info = env.reset() | |
| # env.close() | |
| # fourcc = cv2.VideoWriter_fourcc(*'mp4v') | |
| # out = cv2.VideoWriter('output.mp4', fourcc, 60.0, (rgb_array[0].shape[1], rgb_array[0].shape[0])) | |
| # # Write each frame to the video | |
| # for i in range(len(rgb_array)): | |
| # frame = rgb_array[i][:, :, ::-1] # Convert RGB to BGR | |
| # out.write(frame) | |
| # # Release everything if job is finished | |
| # out.release() | |
| # video_file = open('.\output.mp4', 'rb') | |
| # st.video(video_file.read()) | |
| with open("output.mp4", 'rb') as v: | |
| st.video(v.read()) |