Update app.py

app.py

@@ -16,13 +16,9 @@ def main():
         clicked = st.form_submit_button("Predict Color")
         if clicked:
             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)
-            #result = 'Red' = Red if result[0] == 1 else 'Blue'
+          
+            predicted_Color = predict_color(result)
+           
             st.success('The predicted color is {}'.format(predicted_Color)) 
             
 def predict_Color(prediction):