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() |