Update app.py

app.py

@@ -1,23 +1,23 @@
-import torch
+conda create -n env_pytorch python=3.6
 import streamlit as st
-from PIL import Image
-from megatron import MegatronLM, download_pretrained_model
- 
-#@st.cache(allow_output_mutation=True) # this line is important! without it, the app will crash on first run after a restart. 
-def load_model():
-    model = MegatronLM.from_pretrained('google/megatron-lm-1b')    
-    return model.cuda()
+conda activate env_pytorch
+
 
- 
+import streamlit as st
+from PIL import Image
+import numpy as np
 
- 
+@st.cache(allow_output_mutation=True)
+def load_tileset():
+    tileset = Image.open('nsmbwii_tileset.png')
+    return tileset
 
- 
+ def main():
 
-     #@st.cache(allow_output_mutation=True) # this line is important! without it, the app will crash on first run after a restart. 
+    st.title('NSMBWII Tileset Generator')
 
-     def load_image(filename):        
+    input_tileset = load_tileset()
 
-          image = Image.open(filename).convert("RGB")        
+    st.image(input_tileset, caption='Input Tileset', use_column_width=True)
 
-          return image
+    tile_size = st.slider('Tile Size', min_value=8, max