zamborg commited on
Commit
1371af1
1 Parent(s): 4871a34

propogating cahnges

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -79,12 +79,12 @@ else:
79
  # LOAD AND CACHE THE IMAGE
80
  if uploaded_image is not None:
81
  image = uploaded_image
82
- elif select_idx is None:
83
- image = st.session_state.image
84
  else:
85
  image = Image.open(image_file)
86
 
87
- st.session_state.image = image
88
 
89
  image_dict = imageLoader.transform(image)
90
 
79
  # LOAD AND CACHE THE IMAGE
80
  if uploaded_image is not None:
81
  image = uploaded_image
82
+ elif select_idx is None and 'image' in st.session_state:
83
+ image = st.session_state['image']
84
  else:
85
  image = Image.open(image_file)
86
 
87
+ st.session_state['image'] = image
88
 
89
  image_dict = imageLoader.transform(image)
90