BrunoHenrique00 commited on
Commit
c20bf08
·
unverified ·
1 Parent(s): 7dba675
Files changed (2) hide show
  1. app.py +35 -0
  2. requirements.txt +2 -0
app.py ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ import pandas as pd
3
+ from fastai.tabular.all import *
4
+
5
+ learn = load_learner('model.pkl')
6
+
7
+ def predict(Gender, SmokingHistory, Age, Hypertension, HeartDisease, BMI, HbLevel, GlucoseLevel):
8
+
9
+ data = [[Gender, SmokingHistory, int(Age), int(Hypertension), int(HeartDisease), float(BMI), float(HbLevel), int(GlucoseLevel)]]
10
+
11
+ row_df = pd.DataFrame(data,columns=['gender','smoking_history', 'age', 'hypertension', 'heart_disease', 'bmi', 'HbA1c_level', 'blood_glucose_level'])
12
+
13
+ row, clas, probs = learn.predict(row_df.iloc[0])
14
+ if probs[0] < 0.5:
15
+ return "Tem diabetes"
16
+ else:
17
+ return "Não tem diabetes"
18
+
19
+ gr.Interface(
20
+ fn=predict,
21
+ title="Predict Diabetes",
22
+ allow_flagging="never",
23
+ inputs=[
24
+ gr.inputs.Radio(["Male", "Female"], label="Gender", default="Male"),
25
+ gr.inputs.Dropdown(
26
+ ["never", "No info", "current" , "former"], label="SmokingHistory", default="never"
27
+ ),
28
+ gr.inputs.Number(default=60, label="Age"),
29
+ gr.inputs.Radio(["1", "0"], label="HyperTension", default="1"),
30
+ gr.inputs.Radio(["1", "0"], label="HeartDisease", default="1"),
31
+ gr.inputs.Number(default=30, label="BMI"),
32
+ gr.inputs.Number(default=5.1, label="HbLevel"),
33
+ gr.inputs.Number(default=90, label="GlucoseLevel")
34
+ ],
35
+ outputs="text").launch(inline=False)
requirements.txt ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ fastai
2
+ pandas