Spaces:
Runtime error
Runtime error
Upload do app.py
Browse files
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")
|