|
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) |
|
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) |
|
|
|
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() |
|
|