File size: 1,405 Bytes
050d3e8 652478c 1f5b24d 050d3e8 f6588aa 050d3e8 1cae09b 5df9e7e f6588aa 1cae09b f6588aa 050d3e8 339d7ed 2b32d81 a253179 aa32202 8a5cdbf 4ce5e33 050d3e8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 |
import joblib
import pandas as pd
import streamlit as st
model = joblib.load('model.joblib')
unique_values = joblib.load('unique_values.joblib')
unique_Color = unique_values["Color"]
def main():
st.title("Colors Prediction")
with st.form("questionaire"):
room = st.selectbox("Room", min_value=1, max_value=4)
partition = st.selectbox("Partitions",min_value=1, max_value=3)
rows = st.selectbox("Rows",min_value=1, max_value=2)
columns = st.selectbox("Columns",min_value=1, max_value=2)
clicked = st.form_submit_button("Predict Color")
if clicked:
result = model.predict(pd.DataFrame({"Room": [room],
"Partitions": [partition],
"Rows": [rows],
"Columns": [columns],
"Color": [color]}))
predicted_Color = predict_color(result)
st.success('The predicted color is {}'.format(predicted_Color))
def predict_color(prediction):
if prediction == 'Red':
return 'Red'
elif prediction == 'Blue':
return 'Blue'
elif prediction == 'Yellow':
return 'Yellow'
else:
return 'Green'
if __name__=='__main__':
main()
|