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

Revert "state initialization checks"

Browse files

This reverts commit d2084b7597f12ac582dffcef0d5737346e8bf0fc.

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 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
 
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