Update app.py

app.py

@@ -21,15 +21,8 @@ def predict_intent_to_stay(Engagement, WorkEnvironment, UpperLevelManagement, Re
     # Perform prediction using the loaded model
     prediction = loaded_model.predict_proba(input_data)[0][1]  # Probability of staying
 
-    # Calculate percentage of leaving and staying
-    leave_percentage = (1 - prediction) * 100
-    stay_percentage = prediction * 100
-
-    # Compare predicted value with actual value
-    actual_value = "Stay" if prediction >= 0.5 else "Leave"
-
-    # Return predicted value compared with actual value
-    return {"Predicted": actual_value, "Actual": actual_value}
+    # Return the predicted probability
+    return prediction
 
 # Create sliders for user input
 engagement_slider = gr.Slider(minimum=1, maximum=5, value=3, label="Engagement")
@@ -52,7 +45,7 @@ interface = gr.Interface(
         workload_slider,
         learning_development_slider
     ],
-    outputs="label"
+    outputs=gr.outputs.Label(num_top_classes=2)
 )
 
 # Launch the interface