lucasgbezerra commited on
Commit
133ee94
1 Parent(s): 38834dd

Rename aap to app.py

Browse files
Files changed (2) hide show
  1. aap +0 -0
  2. app.py +35 -0
aap DELETED
File without changes
app.py ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from fastai.tabular.all import *
3
+ import pandas as pd
4
+
5
+ model = load_learner('learn_model.pkl')
6
+
7
+
8
+ def predict(age, hypertension, heart_disease, avg_glucose_level, bmi, gender, married, work_type, residence_type, smoking_status):
9
+
10
+ columns_df = ['gender', 'age', 'hypertension', 'heart_disease', 'ever_married', 'work_type', 'Residence_type', 'avg_glucose_level', 'bmi','smoking_status']
11
+
12
+ data = [[gender, age, hypertension, heart_disease, married, work_type, residence_type, avg_glucose_level, bmi, smoking_status]]
13
+ df = pd.Dataframe(data, columns=columns_df)
14
+
15
+ prediction = model.predict(df)
16
+ return "O paciente tem a seguinte possibilidade de infarto: " + str(predictions[0])
17
+
18
+ gr.Interface(
19
+ fn=predict,
20
+ title="Stroke predict Model",
21
+ allow_flagging="never",
22
+ share=True,
23
+ inputs=[
24
+ gr.inputs.Number(default=30, label="Age"),
25
+ gr.Dropdown(["Yes", "No"], label="Hypertension"),
26
+ gr.Dropdown(["Yes", "No"], label="Heart disease"),
27
+ gr.inputs.Number(default=100, label="Average glucose level"),
28
+ gr.inputs.Number(default=28.8, label="Body mass index (BMI)"),
29
+ gr.inputs.Radio(choices=["Female", "Male", "Other"], default="Female", label="Gender"),
30
+ gr.Dropdown(["No", "Yes"], label="Married"),
31
+ gr.Dropdown(["Governamental", "Never worked", "Private", "Self-employed", "Children"], label="Work type"),
32
+ gr.Dropdown(["Rural", "Urban"], label="Residence type"),
33
+ gr.Dropdown(["Unknown", "Formely smoked", "Never smoked", "Smokes"], label="Smoking status"),
34
+ ],
35
+ outputs="text").launch()