Update app.py

app.py

@@ -250,7 +250,7 @@ with col5: decompose = st.button('Decompose', key="decompose")
 # Make two columns for plots
 _, colA, colB, colC, _ = st.columns([bordersize,1,1,1,bordersize])
 
-if not (image == np.zeros((128,128))).all():
+if not (data == np.zeros((128,128))).all():
     # NORMALIZE IMAGE
     MIN = np.min(np.where(data == 0, 1, data))
     if MIN < 1: data = data / MIN