Update app.py

app.py

@@ -15,7 +15,11 @@ 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)