import streamlit as st @st.cache import sys sys.setrecursionlimit(100000) def load_model(): model = Megatron() return model @st.cache(allow_output_mutation=True) # this line is important! def generate_tileset(model, seed): tileset = model.generate_tileset(seed) # this will take some time to run the first time, but will be cached thereafter return tileset @st.cache(allow_output_mutation=True) def load_tileset(): tileset = Image.open('nsmbwii_tileset.png') return tileset def main(): st.title('NSMBWII Tileset Generator') input_tileset = load_tileset() st.image(input_tileset, caption='Input Tileset', use_column_width=True) tile_size = st.slider('Tile Size', min_value=8, max