File size: 1,288 Bytes
050d3e8 652478c 050d3e8 f6588aa 050d3e8 29babab f6588aa 050d3e8 c743eca 56a85e3 4ce5e33 2924721 aa32202 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 |
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"):
color = st.selectbox("Color", unique_Color)
clicked = st.form_submit_button("Predict Color")
if clicked:
result=model.predict(pd.DataFrame({"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(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()
|