app.py

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