Update app.py (#2)

- Update app.py (4ae7e589686badb7210f306ac995b79b4913f6db)


Co-authored-by: JESUS SANCHEZ <jsr90@users.noreply.huggingface.co>

app.py

@@ -57,12 +57,32 @@ def humands(Sex,Age,Married,Monthlyincome,TotalWorkingYears,DistanceFromHome,Ove
             "JobRole_Healthcare Representative" : [0],
             "EducationField_Human Resources" : [0],
             "JobRole_Manager" : [0],
-            "JobRole_Research Director" : [0],
-            
-            
-                                 
+            "JobRole_Research Director" : [0],              
         }
     )
+
+    columnas = ['Age', 'DailyRate', 'DistanceFromHome', 'Education',
+       'EnvironmentSatisfaction', 'HourlyRate', 'JobInvolvement', 'JobLevel',
+       'JobSatisfaction', 'MonthlyIncome', 'MonthlyRate', 'NumCompaniesWorked',
+       'PercentSalaryHike', 'PerformanceRating', 'RelationshipSatisfaction',
+       'StockOptionLevel', 'TotalWorkingYears', 'TrainingTimesLastYear',
+       'WorkLifeBalance', 'YearsAtCompany', 'YearsInCurrentRole',
+       'YearsSinceLastPromotion', 'YearsWithCurrManager',
+       'BusinessTravel_Non-Travel', 'BusinessTravel_Travel_Frequently',
+       'BusinessTravel_Travel_Rarely', 'Department_Human Resources',
+       'Department_Research & Development', 'Department_Sales',
+       'EducationField_Human Resources', 'EducationField_Life Sciences',
+       'EducationField_Marketing', 'EducationField_Medical',
+       'EducationField_Other', 'EducationField_Technical Degree',
+       'Gender_Female', 'Gender_Male', 'JobRole_Healthcare Representative',
+       'JobRole_Human Resources', 'JobRole_Laboratory Technician',
+       'JobRole_Manager', 'JobRole_Manufacturing Director',
+       'JobRole_Research Director', 'JobRole_Research Scientist',
+       'JobRole_Sales Executive', 'JobRole_Sales Representative',
+       'MaritalStatus_Divorced', 'MaritalStatus_Married',
+       'MaritalStatus_Single', 'OverTime_No', 'OverTime_Yes']
+
+    df = df.reindex(columns=columnas)
          
     pred = model.predict(df)[0]