Spaces:
Sleeping
Sleeping
lucasgbezerra
commited on
Commit
•
133ee94
1
Parent(s):
38834dd
Rename aap to app.py
Browse files
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()
|