ashwml commited on
Commit
9576c97
·
1 Parent(s): 5a4317e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +37 -4
app.py CHANGED
@@ -1,7 +1,40 @@
1
  import gradio as gr
2
 
3
- def greet(name):
4
- return "Hello " + name + "!!"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
5
 
6
- iface = gr.Interface(fn=greet, inputs="text", outputs="text")
7
- iface.launch()
 
1
  import gradio as gr
2
 
3
+ loaded_model="xgboost-model.pkl"
4
+ def predict_death_event(age, anaemia, creatinine_phosphokinase ,diabetes ,ejection_fraction, high_blood_pressure ,platelets ,serum_creatinine, serum_sodium, sex ,smoking ,time):
5
+ input=[[age, anaemia, creatinine_phosphokinase ,diabetes ,ejection_fraction, high_blood_pressure ,platelets ,serum_creatinine, serum_sodium, sex ,smoking ,time]]
6
+ result=loaded_model.predict(input)
7
+
8
+ if result[0]==1:
9
+ return 'Positive'
10
+ else:
11
+ return 'Negative'
12
+ return result
13
+
14
+
15
+
16
+ title = "Patient Survival Prediction"
17
+ description = "Predict survival of patient with heart failure, given their clinical record"
18
+
19
+ out_response = gr.components.Textbox(type="text", label='Death_event')
20
+
21
+ iface = gr.Interface(fn=predict_death_event,
22
+ inputs=[
23
+ gr.Slider(18, 100, value=20, label="Age"),
24
+ gr.Slider(0, 1, value=1, label="anaemia"),
25
+ gr.Slider(100, 2000, value=20, label="creatinine_phosphokinase"),
26
+ gr.Slider(0, 1, value=1, label="diabetes"),
27
+ gr.Slider(18, 100, value=20, label="ejection_fraction"),
28
+ gr.Slider(0, 1, value=1, label="high_blood_pressure"),
29
+ gr.Slider(18, 400000, value=20, label="platelets"),
30
+ gr.Slider(1, 10, value=20, label="serum_creatinine"),
31
+ gr.Slider(100, 200, value=20, label="serum_sodium"),
32
+ gr.Slider(0, 1, value=1, label="sex"),
33
+ gr.Slider(0, 1, value=1, label="smoking"),
34
+ gr.Slider(1, 10, value=20, label="time"),
35
+ ],
36
+ outputs = [out_response])
37
+ iface.launch()
38
+
39
+
40