|
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({"room": [room], |
|
"partitions": [partition], |
|
"Rows": [Row], |
|
"Columns": [Columns], |
|
"Color": [Color]})) |
|
if 'Color' = 'Red' : |
|
print('Red') |
|
elif 'Color' = 'Blue': |
|
print('BLue') |
|
elif 'Color' = 'Yellow': |
|
print('Yellow') |
|
else 'Color' = 'Green': |
|
print('Green') |
|
|
|
st.success('The predicted color is {}'.format(result)) |
|
|
|
if __name__=='__main__': |
|
main() |
|
|