|
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"): |
|
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]})) |
|
if result == 'Red': |
|
print('Red') |
|
elif result == 'Blue': |
|
print('Blue') |
|
elif result == 'Yellow': |
|
print('Yellow') |
|
else: |
|
print('Green') |
|
|
|
st.success('The predicted color is {}'.format(result)) |
|
|
|
if __name__=='__main__': |
|
main() |
|
|