tuestask / app.py
ashwml's picture
Update app.py
38986e6
import gradio as gr
import pickle
import numpy as np
save_file_name="xgboost-model.pkl"
loaded_model = pickle.load(open(save_file_name, 'rb'))
def predict_death_event(age, anaemia, creatinine_phosphokinase ,diabetes ,ejection_fraction, high_blood_pressure ,platelets ,serum_creatinine, serum_sodium, sex ,smoking ,time):
input=[[age, anaemia, creatinine_phosphokinase ,diabetes ,ejection_fraction, high_blood_pressure ,platelets ,serum_creatinine, serum_sodium, sex ,smoking ,time]]
result=loaded_model.predict(input)
if result[0]==1:
return 'Positive'
else:
return 'Negative'
return result
title = "Patient Survival Prediction"
description = "Predict survival of patient with heart failure, given their clinical record"
out_response = gr.components.Textbox(type="text", label='Death_event')
iface = gr.Interface(fn=predict_death_event,
inputs=[
gr.Slider(18, 100, value=20, label="Age"),
gr.Slider(0, 1, value=1, label="anaemia"),
gr.Slider(100, 2000, value=20, label="creatinine_phosphokinase"),
gr.Slider(0, 1, value=1, label="diabetes"),
gr.Slider(18, 100, value=20, label="ejection_fraction"),
gr.Slider(0, 1, value=1, label="high_blood_pressure"),
gr.Slider(18, 400000, value=20, label="platelets"),
gr.Slider(1, 10, value=20, label="serum_creatinine"),
gr.Slider(100, 200, value=20, label="serum_sodium"),
gr.Slider(0, 1, value=1, label="sex"),
gr.Slider(0, 1, value=1, label="smoking"),
gr.Slider(1, 10, value=20, label="time"),
],
outputs = [out_response])
iface.launch()