Update app.py

app.py

@@ -3,7 +3,7 @@ import pandas as pd
 from joblib import load
 
 
-def humands(Monthlyincome,Age,TotalWorkingYears,DistanceFromHome,Overtime,YearsAtCompany,NumCompaniesWorked):
+def humands(Sex,Age,Monthlyincome,TotalWorkingYears,DistanceFromHome,Overtime,YearsAtCompany,NumCompaniesWorked):
  model = load('modelo_entrenado.pkl')
  df = pd.DataFrame.from_dict(
  {
@@ -13,8 +13,8 @@ def humands(Monthlyincome,Age,TotalWorkingYears,DistanceFromHome,Overtime,YearsA
  "DailyRate" : [Monthlyincome*2/30],
  "HourlyRate" : [Monthlyincome*2/1640],
  "DistanceFromHome" : [DistanceFromHome],
- "OverTime_Yes" : [0 if Overtime else 1],
- "OverTime_No" : [0 if not Overtime else 1],
+ "OverTime_Yes" : [1 if Overtime else 0],
+ "OverTime_No" : [1 if not Overtime else 0],
  "YearsAtCompany" : [YearsAtCompany],
  "MonthlyRate" : [Monthlyincome*2],
  "NumCompaniesWorked" : [NumCompaniesWorked],
@@ -38,11 +38,11 @@ def humands(Monthlyincome,Age,TotalWorkingYears,DistanceFromHome,Overtime,YearsA
  "Department_Sales" : [0],
  "JobRole_Laboratory Technician" : [0],
  "Department_Research & Development" : [0],
- "Gender_Female" : [0],
+ "Gender_Female" : [1 if Sex==0 else 0],
  "MaritalStatus_Married" : [0],
  "JobRole_Sales Executive" : [0],
  "EducationField_Technical Degree" : [0],
- "Gender_Male" : [0],
+ "Gender_Male" : [1 if Sex==1 else 0],
  "EducationField_Life Sciences" : [0],
  "BusinessTravel_Travel_Rarely" : [0],
  "MaritalStatus_Divorced" : [0],
@@ -78,8 +78,9 @@ def humands(Monthlyincome,Age,TotalWorkingYears,DistanceFromHome,Overtime,YearsA
 iface = gr.Interface(
  humands,
  [
- gr.inputs.Slider(1000,20000,1,label="Ingresos mensuales del trabajador"),
+ gr.Radio(["Mujer","Hombre"],type = "index",label="Sexo")
  gr.inputs.Slider(18,70,1,label="Edad del trabajador"),
+ gr.inputs.Slider(1000,20000,1,label="Ingresos mensuales del trabajador"),
  gr.inputs.Slider(0,40,1,label="Total de años trabajados del trabajador"),
  gr.inputs.Slider(0,100,1,label="Distancia del trabajo al domicilio en Km"),
  gr.Checkbox(label="¿Realiza horas extras habitualmente?"),