Spaces:
Runtime error
Runtime error
import streamlit as st | |
import pandas as pd | |
import pickle | |
st.title("--Insert title here--") | |
# Step 1 - import saved model | |
model = pickle.load(open("milk_pred.pkl", "rb")) | |
st.write('Insert feature to predict') | |
# Step 2 - prepare input data for user | |
ph = st.slider(label='pH', min_value=3.0, max_value=9.5, value=6.5, step=0.1) | |
temp = st.slider(label='Temprature', min_value=34, max_value=90, value=40, step=1) | |
taste = st.selectbox(label='Taste', options=['Good', 'Bad'], key=1) | |
odor = st.selectbox(label='Odor', options=['Good', 'Bad']) | |
fat = st.selectbox(label='Fat', options=['High', 'Low']) | |
turb = st.selectbox(label='Turbidity', options=['High', 'Low']) | |
color = st.number_input(label='Colour', min_value=240, max_value=255, value=245, step=1) | |
# convert into dataframe | |
data = pd.DataFrame({'pH': [ph], | |
'Temprature': [temp], | |
'Taste': [taste], | |
'Odor':[odor], | |
'Fat': [fat], | |
'Turbidity': [turb], | |
'Colour': [color] | |
}) | |
st.write(data) | |
# model predict | |
clas = model.predict(data).tolist()[0] | |
# interpretation | |
st.write('Classification Result: ') | |
if clas == 0: | |
st.text('Low') | |
elif clas == 1: | |
st.text('Medium') | |
else: | |
st.text('High') | |