first-space / trying.py
Energetic26's picture
output mp4
3b85727
Raw History Blame Contribute Delete
997 Bytes
# import gym
# from IPython import display
# import matplotlib.pyplot as plt
# # %matplotlib inline
# env = gym.make('Breakout-v0')
# env.reset()
# for _ in range(100):
# plt.imshow(env.render(mode='rgb_array'))
# display.display(plt.gcf())
# display.clear_output(wait=True)
# action = env.action_space.sample()
# env.step(action)
import streamlit as st
import numpy as np
import matplotlib.pyplot as plt
import time
import gym
# env = gym.make('CartPole-v1')
# obs = env.reset()
# while True:
# action = env.action_space.sample()
# obs, reward, done, info = env.step(action)
# rgb_array = env.render(mode='rgb_array')
# plt.imshow(env.render(mode='rgb_array'))
# # use the RGB array for further processing or display
# if done:
# break
# Create plot
x = np.linspace(0, 10, 100)
y = np.cos(x)
fig, ax = plt.subplots()
ax.plot(x, y)
ax.set_title("Cosine wave")
container = st.empty()
container.pyplot(fig)
time.sleep(10)
container.empty()