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?"),