Spaces:
Sleeping
Sleeping
| #['preg', 'plas', 'pres', 'skin', 'test', 'mass', 'pedi', 'age', 'class'] | |
| import gradio as gr | |
| import sklearn | |
| import joblib | |
| loaded_model = joblib.load('trained_model.pkl') | |
| def predict(preg, plas, pres, skin, test, mass, pedi, age): | |
| result = loaded_model.predict([[float(preg), float(plas), float(pres), float(skin), int(test), float(mass), float(pedi), float(age)]]) | |
| if int(result[0]) == 1: | |
| return "The person is diabetic" | |
| else: | |
| return "The person is not diabetic" | |
| #return str(result) | |
| demo = gr.Interface( | |
| fn = predict, | |
| inputs = ["text", "text","text", "text","text", "text","text", "text"], | |
| outputs=["text"], | |
| ) | |
| demo.launch() |