Update app.py

app.py

@@ -7,17 +7,13 @@ with open('./salifort_rf3.pickle', 'rb') as file:
 
 # Define the function for making predictions
 def salifort(last_evaluation, number_project, tenure, work_accident, promotion_last_5years, salary, department_IT, department_RandD, department_accounting, department_hr, department_management, department_marketing, department_product_mng, department_sales, department_support, department_technical, overworked):
-    inputs = [['last_evaluation', 'number_project', 'tenure', 'work_accident',
-       'promotion_last_5years', 'salary', 'department_IT', 'department_RandD',
-       'department_accounting', 'department_hr', 'department_management',
-       'department_marketing', 'department_product_mng', 'department_sales',
-       'department_support', 'department_technical', 'overworked']]
-    prediction = model.predict(inputs) 
-    prediction_value = prediction[0][0]
+    inputs = [[float(last_evaluation), float(number_project), float(tenure), float(work_accident), float(promotion_last_5years), float(salary), float(department_IT), float(department_RandD), float(department_accounting), float(department_hr), float(department_management), float(department_marketing), float(department_product_mng), float(department_sales), float(department_support), float(department_technical), float(overworked)]]
+    prediction = model.predict(inputs)
+    prediction_value = prediction[0]
     if prediction_value == 0:
-        label_text = 'employee would not leave the company'
+        label_text = 'Employee would not leave the company.'
     else:
-        label_text = 'employee will leave the company'
+        label_text = 'Employee will leave the company.'
     return label_text
 
 # Create the Gradio interface