Update app.py

app.py

@@ -126,24 +126,23 @@ app_mode = st.sidebar.selectbox("Please select from the following", SIDEBAR_OPTI
 photos = ["fight.jpg","shaolin-kung-fu.jpg"]
 clr = [(185, 39, 40),(40, 96, 219),(249, 223, 2)]
 colour = ["Red","Blue","Yellow"]
-if app_mode == SIDEBAR_OPTION_DEMO_IMAGE and app_mode == SIDEBAR_OPTION_COLOUR_IMAGE:
+
+if app_mode == SIDEBAR_OPTION_DEMO_IMAGE:
   st.sidebar.write(" ------ ")
-  option = st.sidebar.selectbox('Please select a sample image, then click Magic Time button', photos)
   option2 = st.sidebar.selectbox('Please select a sample image, then click Magic Time button', colour)
   colour[0] = clr[0]
   colour[1] = clr[1]
   colour[2] = clr[2]
+if app_mode == SIDEBAR_OPTION_DEMO_IMAGE:
+  st.sidebar.write(" ------ ")
+  option = st.sidebar.selectbox('Please select a sample image, then click Magic Time button', photos)
   pressed = st.sidebar.button('PoP')
   if pressed:
      st.empty()
      st.sidebar.write('Please wait for the magic to happen! This may take up to a minute.')
      predict(option,option2)
 
-elif app_mode == SIDEBAR_OPTION_UPLOAD_IMAGE and app_mode == SIDEBAR_OPTION_COLOUR_IMAGE:
+elif app_mode == SIDEBAR_OPTION_UPLOAD_IMAGE:
   uploaded_file = st.file_uploader("Choose an image...")
-  option2 = st.sidebar.selectbox('Please select a sample image, then click Magic Time button', colour)
-  colour[0] = clr[0]
-  colour[1] = clr[1]
-  colour[2] = clr[2]
   if uploaded_file is not None:
     predict(uploaded_file,option2)