Update app.py

app.py

@@ -5,17 +5,13 @@ import streamlit as st
 model = joblib.load('model.joblib')
 unique_values = joblib.load('unique_values.joblib')
 
-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")
 
     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: