Update app.py

app.py

@@ -11,21 +11,21 @@ def main():
     st.title("Colors Prediction")
 
     with st.form("questionaire"):
-        color = st.selectbox("Color", unique_Color)
+        Color = st.selectbox("Color", unique_Color)
         
         clicked = st.form_submit_button("Predict Color")
         if clicked:
-            result=model.predict(pd.DataFrame({"Color":[color]}))
+            result=model.predict(pd.DataFrame({"Color":[Color]}))
             #result=model.predict(pd.DataFrame({"Room": [room],
             #                                   "Partitions": [partition],
             #                                   "Rows": [rows],
             #                                   "Columns": [columns],
             #                                   "Color": [color]}))
-            predicted_color = predict_color(result)
+            predicted_Color = predict_Color(result)
             #result = 'Red' = Red if result[0] == 1 else 'Blue'
             st.success('The predicted color is {}'.format(predicted_color)) 
             
-def predict_color(prediction):
+def predict_Color(prediction):
     if prediction == 'Red':
         return 'Red'
     elif prediction == 'Blue':