Vivekan commited on
Commit
38332b6
1 Parent(s): 40fc269

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -1,7 +1,9 @@
1
  from sklearn.linear_model import LogisticRegression
2
-
3
  LR=LogisticRegression()
4
 
 
 
5
 
6
  def classify(num):
7
  if num<1:
@@ -12,7 +14,7 @@ import gradio as gr
12
  import numpy as np
13
  def predict_diabetes(preg,glu,bp,st,ins,bmi,dpf,age):
14
  input_array=np.array([[preg,glu,bp,st,ins,bmi,dpf,age]])
15
- pred=LR.predict(input_array)
16
  output=classify(pred[0])
17
  if output=='negative':
18
  return [(0,output)]
 
1
  from sklearn.linear_model import LogisticRegression
2
+ import pickle
3
  LR=LogisticRegression()
4
 
5
+ pickle.dump(LR,open('model.pkl','wb'))
6
+ model = pickle.load(open('model.pkl','rb'))
7
 
8
  def classify(num):
9
  if num<1:
 
14
  import numpy as np
15
  def predict_diabetes(preg,glu,bp,st,ins,bmi,dpf,age):
16
  input_array=np.array([[preg,glu,bp,st,ins,bmi,dpf,age]])
17
+ pred=model.predict(input_array)
18
  output=classify(pred[0])
19
  if output=='negative':
20
  return [(0,output)]