test / app.py
Galatea007's picture
Create app.py
e97d334 verified
raw
history blame contribute delete
842 Bytes
import streamlit as st
import pandas as pd
import pickle
# Load Model
model = pickle.load(open('logreg_model.pkl', 'rb'))
st.title('Iris Variety Prediction')
# Form
with st.form(key='form_parameters'):
sepal_length = st.slider('Sepal Length', 4.0, 8.0, 4.0)
sepal_width = st.slider('Sepal Width', 2.0, 4.5, 2.0)
petal_length = st.slider('Petal Length', 1.0, 7.0, 1.0)
petal_width = st.slider('Petal Width', 0.1, 2.5, 0.1)
st.markdown('---')
submitted = st.form_submit_button('Predict')
# Data Inference
data_inf = {
'sepal.length': sepal_length,
'sepal.width': sepal_width,
'petal.length': petal_length,
'petal.width': petal_width
}
data_inf = pd.DataFrame([data_inf])
if submitted:
# Predict using Logistic Regression
y_pred_inf = model.predict(data_inf)
st.write('## Iris Variety = '+ str(y_pred_inf))