app

app.py

@@ -27,10 +27,10 @@ def update(array_value):
     # In future pass data through array_value parameter
     if array_value == "2,2,1,3,1,2,0,1,40,1":
         prediction_value = reg.predict(np.array([[2,2,1,3,1,2,0,1,40,1]]))
-        return f"Prediction : {prediction_value}!"
+        return f"Attrition Prediction : {prediction_value}!"
     if array_value == "0,0,0,3,0,2,0,1,2,1":
         prediction_value = reg.predict(np.array([[0,0,0,3,0,2,0,1,2,1]]))
-        return f"Prediction : {prediction_value}!"
+        return f"Attrition Prediction : {prediction_value}!"
 
 with gr.Blocks() as demo:
     gr.Markdown("*** Employee Experience Prediction ***")