Graziela commited on
Commit
899f195
·
1 Parent(s): af788c5

Upload do app.py

Browse files
Files changed (1) hide show
  1. app.py +32 -0
app.py ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # -*- coding: utf-8 -*-
2
+ """app.ipynb
3
+
4
+ Automatically generated by Colaboratory.
5
+
6
+ Original file is located at
7
+ https://colab.research.google.com/drive/1I-y23tTTs_aMjSYwPa3Fk8Ywz8H8ZUgl
8
+ """
9
+
10
+ import gradio as gr
11
+ import joblib as jb
12
+
13
+ def predict(sex, age, hypertension, heart_disease, ever_married, work_type, Residence_type, avg_glucose_level, bmi, smoking_status):
14
+ model = jb.load('model.pkl')
15
+
16
+ p = model.predict_proba({{sex, age, hypertension, heart_disease, ever_married, work_type, Residence_type, avg_glucose_level, bmi, smoking_status}})[0]
17
+
18
+ return {"Não terá AVC": p[0], "Terá AVC": p[1]}
19
+
20
+ demo = gr.Interface(fn=predict,
21
+ inputs=[gr.Dropdown(choices=["female", "male"], type="index"),
22
+ "number",
23
+ gr.Dropdown(choices=["Não Hipertenso", "Hipertenso"], type="index"),
24
+ gr.Dropdown(choices=["Não doença cardíaca", "Possui doença cardíaca"], type="index"),
25
+ gr.Dropdown(choices=["Não casado(a)", "Casado(a)"], type="index"),
26
+ gr.Dropdown(choices=["Nunca trabalhou", "filhos", "Emprego público", "Autônomo", "Privado"], type="index"),
27
+ gr.Dropdown(choices=["Rural", "Urbana"], type="index"),
28
+ "number",
29
+ "number",
30
+ gr.Dropdown(choices=["Nunca fumou", "Fuma"], type="index"),
31
+ ],
32
+ outputs="label")