Update app.py

app.py

@@ -23,14 +23,15 @@ def main():
         
         clicked = st.form_submit_button("Predict Color")
         if clicked:
-            result=model.predict(pd.DataFrame({"Room": [Room],
-                                               "Partitions": [Partition],
-                                               "Rows": [Rows],
-                                               "Columns": [Columns],
-                                               "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'
-            st.success('The predicted color is {}'.format(result))
+            st.success('The predicted color is {}'.format(predicted_color)) 
+            
 def predict_color(prediction):
     if prediction == 'Red':
         return 'Red'