Test #A

app.py

@@ -1,11 +1,15 @@
-conda create -n env_pytorch python=3.6
 import streamlit as st
-conda activate env_pytorch
 
+@st.cache
+def load_model():
+    model = Megatron()
+    return model
 
-import streamlit as st
-from PIL import Image
-import numpy as np
+@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():