Spaces:
Runtime error
Runtime error
import gradio as gr | |
import skimage | |
import pickle | |
import pandas as pd | |
with open('model.pkl', 'rb') as f: | |
model = pickle.load(f) | |
def predict(Pregnancies, Glucose, BloodPressure, SkinThickness, Insulin, BMI, DiabetesPedigreeFunction, Age): | |
data = [[int(Pregnancies), int(Glucose), int(BloodPressure), int(SkinThickness), int(Insulin), float(BMI), float(DiabetesPedigreeFunction), int(Age)]] | |
row_df=pd.DataFrame(data,columns=['Pregnancies', 'Glucose', 'BloodPressure', 'SkinThickness','Insulin', 'BMI', 'DiabetesPedigreeFunction', 'Age']) | |
predictions = model.predict(row_df) | |
y_pred = model.predict(row_df) | |
if y_pred[0] == 1: | |
return "Tem diabetes" | |
else: | |
return "Não tem diabetes" | |
return 0 | |
gr.Interface( | |
fn=predict, | |
title="Predict Diabetes", | |
allow_flagging="never", | |
inputs=[ | |
gr.inputs.Number(default=1, label="Pregnancies"), | |
gr.inputs.Number(default=126, label="Glucose"), | |
gr.inputs.Number(default=60, label="BloodPressure"), | |
gr.inputs.Number(default=0, label="SkinThickness"), | |
gr.inputs.Number(default=0, label="Insulin"), | |
gr.inputs.Number(default=30.1, label="BMI"), | |
gr.inputs.Number(default=0.349, label="DiabetesPedigreeFunction"), | |
gr.inputs.Number(default=47, label="Age") | |
], | |
outputs="text").launch() |