pjeena/Testing_iris
0
1import streamlit as st2import pandas as pd3import pickle4# Load Model5model = pickle.load(open('logreg_model.pkl', 'rb'))6st.title('Iris Variety Prediction')7# Form8with st.form(key='form_parameters'):9 sepal_length = st.slider('Sepal Length', 4.0, 8.0, 4.0)10 sepal_width = st.slider('Sepal Width', 2.0, 4.5, 2.0)11 petal_length = st.slider('Petal Length', 1.0, 7.0, 1.0)12 petal_width = st.slider('Petal Width', 0.1, 2.5, 0.1)13 st.markdown('---')14 submitted = st.form_submit_button('Predict')15# Data Inference16data_inf = {17 'SepalLengthCm': sepal_length,18 'SepalWidthCm': sepal_width,19 'PetalLengthCm': petal_length,20 'PetalWidthCm': petal_width21}22data_inf = pd.DataFrame([data_inf])23if submitted:24 # Predict using Logistic Regression25 y_pred_inf = model.predict(data_inf)26 st.write('## Iris Variety = '+ str(y_pred_inf))