Update app.py

app.py

@@ -129,9 +129,11 @@ 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)
   colour = st.sidebar.selectbox("Colour", colour)
-  colour[0] = (185, 39, 40)
-  colour[1] = (40, 96, 219)
-  colour[2] = (249, 223, 2)
+  if colour == 'Red':
+    colour = (185, 39, 40)
+  #colour[0] = (185, 39, 40)
+  #colour[1] = (40, 96, 219)
+  #colour[2] = (249, 223, 2)
   pressed = st.sidebar.button('PoP')
   if pressed:
      st.empty()
@@ -142,7 +144,9 @@ elif app_mode == SIDEBAR_OPTION_UPLOAD_IMAGE:
   uploaded_file = st.file_uploader("Choose an image...")
   if uploaded_file is not None:
     colour = st.sidebar.selectbox("Colour", colour)
-    colour[0] = (185, 39, 40)
-    colour[1] = (40, 96, 219)
-    colour[2] = (249, 223, 2)
+    if colour == 'Red':
+    colour = (185, 39, 40)
+    #colour[0] = (185, 39, 40)
+    #colour[1] = (40, 96, 219)
+    #colour[2] = (249, 223, 2)
     predict(uploaded_file,colour)