File size: 1,582 Bytes
050d3e8 652478c 4ce5e33 652478c 050d3e8 f6588aa 050d3e8 4ce5e33 29babab f6588aa 050d3e8 652478c f6588aa 4ce5e33 2924721 f6588aa 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 45 46 |
import joblib
import pandas as pd
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"):
room = st.selectbox("Room", unique_Room)
partition = st.selectbox("Partitions", unique_Partitions)
rows = st.selectbox("Rows", unique_Rows)
columns = st.selectbox("Columns", unique_Columns)
color = st.selectbox("Color", unique_Color)
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)
#result = 'Red' = Red if result[0] == 1 else 'Blue'
st.success('The predicted color is {}'.format(result))
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()
|