bbhagat1 commited on
Commit
97ba311
1 Parent(s): 8b5557b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +39 -15
app.py CHANGED
@@ -2,23 +2,47 @@ import pandas as pd
2
  import pickle
3
  import numpy as np
4
  from sklearn import tree
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
5
  # Load the Random Forest CLassifier model
6
  filename = 'model.pkl'
7
  loaded_model = pickle.load(open(filename, 'rb'))
8
  print(loaded_model)
9
 
10
- input = [0.0,0.0,1.0,26.0,1.0,0.0,0.0,0.0,0.0,1.0,0.0,1.0,0.0,3.0,0.0,15.0,0.0,0.0,7.0,5.0,7.0]
11
- manualInput =[
12
- [1.0,1.0,1.0,37.0,1.0,1.0,1.0,0.0,0.0,1.0,0.0,1.0,0.0,5.0,0.0,0.0,1.0,1.0,10.0,6.0,5.0]
13
- #,[1.0,1.0,1.0,28.0,1.0,0.0,1.0,0.0,0.0,1.0,0.0,1.0,0.0,4.0,0.0,0.0,0.0,1.0,12.0,2.0,4.0]
14
- #,[1.0,1.0,1.0,27.0,1.0,0.0,0.0,0.0,1.0,1.0,0.0,1.0,0.0,4.0,20.0,20.0,1.0,0.0,8.0,4.0,7.0]
15
- ]
16
- print(input, manualInput)
17
- col=["HighBP","HighChol","CholCheck","BMI","Smoker","Stroke","HeartDiseaseorAttack","PhysActivity","Fruits"
18
- ,"Veggies","HvyAlcoholConsump","AnyHealthcare","NoDocbcCost","GenHlth","MentHlth","PhysHlth","DiffWalk"
19
- ,"Sex","Age","Education","Income"]
20
- ddf = pd.DataFrame(manualInput, columns=col)
21
- print(ddf)
22
-
23
- result = loaded_model.predict(ddf)
24
- print(result)
 
2
  import pickle
3
  import numpy as np
4
  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
+
43
+ result = classifier.predict(multiple_records)
44
+ print(result)
45
+
46
+
47
+ iface = gr.Interface(fn=predict2, inputs="text", outputs="text")
48
+ iface.launch()