Galatea007 commited on
Commit
e97d334
1 Parent(s): fb778ee

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +26 -0
app.py ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ import pandas as pd
3
+ import pickle
4
+ # Load Model
5
+ model = pickle.load(open('logreg_model.pkl', 'rb'))
6
+ st.title('Iris Variety Prediction')
7
+ # Form
8
+ with 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 Inference
16
+ data_inf = {
17
+ 'sepal.length': sepal_length,
18
+ 'sepal.width': sepal_width,
19
+ 'petal.length': petal_length,
20
+ 'petal.width': petal_width
21
+ }
22
+ data_inf = pd.DataFrame([data_inf])
23
+ if submitted:
24
+ # Predict using Logistic Regression
25
+ y_pred_inf = model.predict(data_inf)
26
+ st.write('## Iris Variety = '+ str(y_pred_inf))