Update app.py

app.py

@@ -4,8 +4,12 @@ import streamlit as st
 
 model = joblib.load('model.joblib')
 unique_values = joblib.load('unique_values.joblib')
-    
-unique_color =  unique_values["color"]
+
+unique_Room =  unique_values["Room"]
+unique_Partitions =  unique_values["Partitions"]
+unique_Rows =  unique_values["Rows"]
+unique_Columns=  unique_values["Columns"]
+unique_Color =  unique_values["Color"]
 
 def main():
     st.title("Colors Prediction")
@@ -15,9 +19,9 @@ def main():
         
         clicked = st.form_submit_button("Predict Color")
         if clicked:
-            result=model.predict(pd.DataFrame({"room": [room],
-                                               "partitions": [partition],
-                                               "Rows": [Row],
+            result=model.predict(pd.DataFrame({"Room": [Room],
+                                               "Partitions": [Partition],
+                                               "Rows": [Rows],
                                                "Columns": [Columns],
                                                "Color": [Color]}))