Spaces:
Sleeping
Sleeping
Galatea007
commited on
Commit
•
e97d334
1
Parent(s):
fb778ee
Create app.py
Browse files
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))
|