memory lose fixed

app.py

@@ -5,6 +5,9 @@ import torch
 from utils import *
 import matplotlib.pyplot as plt
 
+if not hasattr(st, 'paths'):
+    st.paths = None
+
 # Load Model
 # @title Load pretrained weights
 
@@ -54,6 +57,7 @@ st.markdown('or use a demo sample')
 if st.button('sample 1'):
     sample_path = 'demo_data/lombardia'
 
+paths = None
 if sample_path is not None:
     st.markdown(f'elaborating {sample_path}...')
 
@@ -145,9 +149,12 @@ if sample_path is not None:
     st.markdown('End prediction')
 
     folder = "demo_data/results/seg_maps/example-lombardia/2"
-    paths = os.listdir(folder)
+    st.paths = os.listdir(folder)
 
-    file_picker = st.selectbox("Select day predict (annual is patch-pred-nn.tif)", paths, index=paths.index('patch-pred-nn.tif'))
+if st.paths is not None:
+    folder = "demo_data/results/seg_maps/example-lombardia/2"
+    file_picker = st.selectbox("Select day predict (annual is patch-pred-nn.tif)",
+                               st.paths, index=st.paths.index('patch-pred-nn.tif'))
 
     file_path = os.path.join(folder, file_picker)
     print(file_path)
@@ -158,4 +165,3 @@ if sample_path is not None:
     fig, ax = plt.subplots()
     ax.imshow(target)
     st.pyplot(fig)
-