Reyne commited on
Commit
08459ab
·
1 Parent(s): 69e837c

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +61 -0
app.py ADDED
@@ -0,0 +1,61 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from joblib import load
2
+ import gradio as gr
3
+ import pandas as pd
4
+
5
+ from model_training.log_regress import *
6
+
7
+ clf = load("model_training/regressao_logistica.joblib")
8
+
9
+ def passou_ou_nao(sexo, ano_nasc, raca, renda, auxilios, grau,area, da_capital):
10
+
11
+ auxilios_list=3*["False"]
12
+
13
+ for i in auxilios:
14
+ #print(i)
15
+ auxilios_list[i] = "True"
16
+
17
+ aux_ali, aux_Mor, aux_Trans = auxilios_list
18
+ entrada = [[sexo[0],ano_nasc,raca,renda, aux_ali, aux_Trans, aux_Mor, grau,area,da_capital]]
19
+
20
+ headers = ['sexo', 'ano_nascimento', 'raca', 'renda', 'possui_auxilio_alimentacao',
21
+ 'possui_auxilio_transporte', 'possui_auxilio_residencia_moradia',
22
+ 'grau_academico', 'area_conhecimento', 'local_ou_de_fora']
23
+
24
+ entrada_df = pd.DataFrame(entrada,columns=headers)
25
+
26
+ prediction = clf.predict_proba(entrada_df)[0]
27
+
28
+ class_names = ["REPROVAR", "APROVADO"]
29
+
30
+ return {class_names[i]: prediction[i] for i in range(2)}
31
+
32
+ #set the user uploaded image as the input array
33
+ #match same shape as the input shape in the model
34
+
35
+ #setup the interface
36
+ iface = gr.Interface(
37
+ passou_ou_nao,
38
+ [
39
+ gr.inputs.Radio(["Masculino", "Feminino"]),
40
+ gr.inputs.Number(label="Ano de nascimento"),
41
+ gr.inputs.Radio(['Negro', 'Branco', 'Pardo',
42
+ 'Amarelo (de origem oriental)', 'Indígena',
43
+ 'Remanescente de quilombo']),
44
+ gr.inputs.Number(label="Sua Renda familiar"),
45
+ gr.inputs.CheckboxGroup( ["Alimentação","Moradia/Residência","Transporte"],
46
+ label="Possui Auxilio ... ?",type='index'),
47
+
48
+ gr.inputs.Dropdown(['BACHARELADO', 'LICENCIATURA', 'TECNOLÓGICO'], label="Nível"),
49
+ gr.inputs.Dropdown(['Ciências Biológicas', 'Ciências da Saúde',
50
+ 'Ciências Exatas e da Terra', 'Engenharias',
51
+ 'Ciências Sociais Aplicadas', 'Ciências Humanas',
52
+ 'Linguística, Letras e Artes', 'Ciências Agrárias', 'Outra'],
53
+ label="Sua Aréa de conhecimento"),
54
+ gr.inputs.Checkbox(default=False, label="Sou originário da capital do RN"),
55
+ ],
56
+
57
+ outputs = gr.outputs.Label(),
58
+ interpretation="default"
59
+ )
60
+
61
+ iface.launch()