File size: 1,320 Bytes
050d3e8 652478c 050d3e8 f6588aa 050d3e8 f6588aa 050d3e8 652478c f6588aa d858b13 2924721 f6588aa 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_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')
#result = 'Red' = Red if result[0] == 1 else 'Blue'
st.success('The predicted color is {}'.format(result))
if __name__=='__main__':
main()
|