bbhagat1 commited on
Commit
242a3da
1 Parent(s): 97ba311

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -16
app.py CHANGED
@@ -5,38 +5,33 @@ from sklearn import tree
5
 
6
  import gradio as gr
7
 
 
 
 
 
 
8
 
9
  def multiline(textData):
10
  print("inp", textData)
11
  col=["HighBP","HighChol","CholCheck","BMI","Smoker","Stroke","HeartDiseaseorAttack","PhysActivity","Fruits"
12
  ,"Veggies","HvyAlcoholConsump","AnyHealthcare","NoDocbcCost","GenHlth","MentHlth","PhysHlth","DiffWalk"
13
  ,"Sex","Age","Education","Income"]
14
- empty_array = []
 
15
  for line in textData.split("\n"):
16
  abc = list(map(float, line.split(",")));
17
  print(abc)
18
- empty_array = np.append(empty_array, np.array([abc]), axis=0)
 
19
  print(empty_array)
20
  ddf = pd.DataFrame(empty_array, columns=col)
 
21
  print(ddf)
22
 
23
- print(loaded_model.predict(ddf))
24
  return ddf
25
 
26
 
27
-
28
-
29
-
30
-
31
-
32
-
33
-
34
- # Load the Random Forest CLassifier model
35
- filename = 'model.pkl'
36
- loaded_model = pickle.load(open(filename, 'rb'))
37
- print(loaded_model)
38
-
39
-
40
  def predict2(content):
41
  multiple_records = multiline(content)
42
 
 
5
 
6
  import gradio as gr
7
 
8
+ # Load the Random Forest CLassifier model
9
+ filename = 'model.pkl'
10
+ loaded_model = pickle.load(open(filename, 'rb'))
11
+ print(loaded_model)
12
+
13
 
14
  def multiline(textData):
15
  print("inp", textData)
16
  col=["HighBP","HighChol","CholCheck","BMI","Smoker","Stroke","HeartDiseaseorAttack","PhysActivity","Fruits"
17
  ,"Veggies","HvyAlcoholConsump","AnyHealthcare","NoDocbcCost","GenHlth","MentHlth","PhysHlth","DiffWalk"
18
  ,"Sex","Age","Education","Income"]
19
+ #empty_array = []
20
+ empty_array = np.empty((0, 21), float)
21
  for line in textData.split("\n"):
22
  abc = list(map(float, line.split(",")));
23
  print(abc)
24
+ empty_array = np.append(empty_array, np.array([abc]), axis=0)
25
+ print("empty_array")
26
  print(empty_array)
27
  ddf = pd.DataFrame(empty_array, columns=col)
28
+ print("ddf")
29
  print(ddf)
30
 
31
+ #print(loaded_model.predict(ddf))
32
  return ddf
33
 
34
 
 
 
 
 
 
 
 
 
 
 
 
 
 
35
  def predict2(content):
36
  multiple_records = multiline(content)
37