Spaces:
Runtime error
Runtime error
File size: 1,572 Bytes
4e988b2 40a8579 ac709a4 4e988b2 57f697a 4e988b2 7e2d2ed 4e988b2 67c6faa 4e988b2 ca8fe9c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 |
import pickle
import gradio as gr
import numpy as np
import pandas as pd
from sklearn.preprocessing import StandardScaler
from sklearn.linear_model import LogisticRegression
# import sklearn.preprocessing
# with open("diabetes_classifier.pkl", "rb") as file:
# loaded_model = pickle.load(file)
loaded_model = pickle.load(open("diabetes_classifier.pkl", "rb"), encoding="bytes")
diabetes_classifier = loaded_model['model']
columns = loaded_model['columns']
def predict_diabetes_func(Pregnancies, Glucose, BloodPressure, SkinThickness, Insulin, BMI, DiabetesPedigreeFunction, Age):
input_data = [Pregnancies, Glucose, BloodPressure, SkinThickness, Insulin, BMI, DiabetesPedigreeFunction, Age]
input_df = pd.DataFrame([input_data], columns=columns)
prediction = diabetes_classifier.predict(input_df)
# return Pregnancies
return "Positive" if prediction[0] == 1 else "Negative"
iface = gr.Interface( title = "Mashdemy AI Demo _Diabetes Prediction App",
description = "Enter the various parameters and click submit to know if the result is Positive or Negative",
fn=predict_diabetes_func, # Updated function name
inputs=[
gr.Number(label="Pregnancies"),
gr.Number(label="Glucose"),
gr.Number(label="BloodPressure"),
gr.Number(label="SkinThickness"),
gr.Number(label="Insulin"),
gr.Number(label="BMI"),
gr.Number(label="DiabetesPedigreeFunction"),
gr.Number(label="Age"),
],
outputs="text",
live=False,
)
iface.launch(share= True, debug = True) |