Update app.py

app.py

@@ -30,8 +30,9 @@ def load_dataset():
 
 @st.experimental_singleton
 def load_variables():# Don't want to open read files over and over. not sure if it makes a diff
-    st.session_state['latent_walk_code']=open("assets/code_snippets/latent_walk.py").read()
-    st.session_state['latent_walk_code_music']=open("assets/code_snippets/latent_walk_music.py").read()
+    latent_walk_code=open("assets/code_snippets/latent_walk.py").read()
+    latent_walk_code_music=open("assets/code_snippets/latent_walk_music.py").read()
+    return latent_walk_code,latent_walk_code_music
 
 def img2download(image):
         imgByteArr = io.BytesIO()
@@ -44,7 +45,7 @@ model_version='57d36a15546909557d9f967f47713236c8288838'
 # model_version=None
 model=load_model_intocache(model_name,model_version)
 dataset=load_dataset()
-load_variables()
+latent_walk_code, latent_walk_code_music=load_variables()
 
 generate_menu="🦋 Make butterflies"
 latent_walk_menu="🎧 Take a latent walk"
@@ -90,8 +91,6 @@ Based on [paper:](https://openreview.net/forum?id=1Fqg133qRaI) *Towards Faster a
 
 elif screen == latent_walk_menu:
 
-    latent_walk_code=open("assets/code_snippets/latent_walk.py").read()
-    latent_walk_music_code=open("assets/code_snippets/latent_walk_music.py").read()
     st.write("Take a latent walk :musical_note: with cute butterflies")
 
     cols=st.columns(3)
@@ -108,9 +107,9 @@ elif screen == latent_walk_menu:
     st.write("🎧Did those butterflies seem to be dancing to the music?!Here is the secret:")
     with st.expander("See the Code Snippets"):
         st.write("A regular latent walk:")
-        st.code(st.session_state['latent_walk_code'], language='python')
+        st.code(latent_walk_code, language='python')
         st.write(":musical_note: latent walk with music:")
-        st.code(st.session_state['latent_walk_code_music'], language='python')
+        st.code(latent_walk_code_music, language='python')
 
 
 elif screen == make_meme_menu: