Update app.py
Browse files
app.py
CHANGED
@@ -15,7 +15,11 @@ def main():
|
|
15 |
|
16 |
clicked = st.form_submit_button("Predict Color")
|
17 |
if clicked:
|
18 |
-
result=model.predict(pd.DataFrame({"
|
|
|
|
|
|
|
|
|
19 |
|
20 |
predicted_Color = predict_color(result)
|
21 |
|
|
|
15 |
|
16 |
clicked = st.form_submit_button("Predict Color")
|
17 |
if clicked:
|
18 |
+
result = model.predict(pd.DataFrame({"Room": [room],
|
19 |
+
"Partitions": [partition],
|
20 |
+
"Rows": [rows],
|
21 |
+
"Columns": [columns],
|
22 |
+
"Color": [color]}))
|
23 |
|
24 |
predicted_Color = predict_color(result)
|
25 |
|