Spaces:
Runtime error
Runtime error
DanielViniciusAlves
commited on
Commit
•
239eec5
1
Parent(s):
8f641b5
Update app.py
Browse files
app.py
CHANGED
@@ -7,31 +7,32 @@ import pandas as pd
|
|
7 |
with open('model.pkl', 'rb') as f:
|
8 |
model = pickle.load(f)
|
9 |
|
10 |
-
def predict(
|
11 |
|
12 |
-
data = [[
|
13 |
|
14 |
-
row_df=pd.DataFrame(data,columns=['
|
15 |
|
16 |
predictions = model.predict(row_df)
|
17 |
-
|
|
|
|
|
|
|
|
|
|
|
18 |
|
19 |
gr.Interface(
|
20 |
fn=predict,
|
21 |
-
title="
|
22 |
allow_flagging="never",
|
23 |
inputs=[
|
24 |
-
gr.inputs.Number(default=
|
25 |
-
gr.inputs.Number(default=
|
26 |
-
gr.inputs.Number(default=
|
27 |
-
gr.inputs.Number(default=
|
28 |
-
gr.inputs.Number(default=0
|
29 |
-
gr.inputs.Number(default=
|
30 |
-
gr.inputs.Number(default=
|
31 |
-
gr.inputs.Number(default=
|
32 |
-
gr.inputs.Number(default=3.5, label="pH"),
|
33 |
-
gr.inputs.Number(default=0.6, label="sulphates"),
|
34 |
-
gr.inputs.Number(default=9.5, label="alcohol"),
|
35 |
-
gr.inputs.Number(default=1.6, label="LogTotalSulfurDioxide"),
|
36 |
],
|
37 |
outputs="text").launch()
|
|
|
7 |
with open('model.pkl', 'rb') as f:
|
8 |
model = pickle.load(f)
|
9 |
|
10 |
+
def predict(Pregnancies, Glucose, BloodPressure, SkinThickness, Insulin, BMI, DiabetesPedigreeFunction, Age):
|
11 |
|
12 |
+
data = [[int(Pregnancies), int(Glucose), int(BloodPressure), int(SkinThickness), int(Insulin), float(BMI), float(DiabetesPedigreeFunction), int(Age)]]
|
13 |
|
14 |
+
row_df=pd.DataFrame(data,columns=['Pregnancies', 'Glucose', 'BloodPressure', 'SkinThickness','Insulin', 'BMI', 'DiabetesPedigreeFunction', 'Age'])
|
15 |
|
16 |
predictions = model.predict(row_df)
|
17 |
+
y_pred = model.predict(row_df)
|
18 |
+
if y_pred[0] == 1:
|
19 |
+
return "Tem diabetes"
|
20 |
+
else:
|
21 |
+
return "Não tem diabetes"
|
22 |
+
return 0
|
23 |
|
24 |
gr.Interface(
|
25 |
fn=predict,
|
26 |
+
title="Predict Diabetes",
|
27 |
allow_flagging="never",
|
28 |
inputs=[
|
29 |
+
gr.inputs.Number(default=1, label="Pregnancies"),
|
30 |
+
gr.inputs.Number(default=126, label="Glucose"),
|
31 |
+
gr.inputs.Number(default=60, label="BloodPressure"),
|
32 |
+
gr.inputs.Number(default=0, label="SkinThickness"),
|
33 |
+
gr.inputs.Number(default=0, label="Insulin"),
|
34 |
+
gr.inputs.Number(default=30.1, label="BMI"),
|
35 |
+
gr.inputs.Number(default=0.349, label="DiabetesPedigreeFunction"),
|
36 |
+
gr.inputs.Number(default=47, label="Age")
|
|
|
|
|
|
|
|
|
37 |
],
|
38 |
outputs="text").launch()
|