app update

app.py

@@ -66,8 +66,11 @@ def predict(SeniorCitizen, Partner, Dependents, tenure,
     final_df = pd.concat([remaining_columns, first_three_columns], axis=1)
 
     # Make predictions using the model
-    prediction = model.predict(final_df)[0]
-    prediction_label = "Customer is likely to CHURN 🔴" if prediction == "Yes" else "Customer is NOT likely to CHURN ✅"
+    prediction_probs = model.predict_proba(final_df)[0]
+    prediction_label = {
+        "Prediction: CHURN 🔴": prediction_probs[1],
+        "Prediction: STAY ✅": prediction_probs[0]
+    }
 
     return prediction_label
 
@@ -136,9 +139,5 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
                     """)
         
     predict_btn.click(fn=predict, inputs=input_interface, outputs=output_interface)
-    
-app.launch(share=True, server_port=8080, debug=True)
-
-# Include the custom CSS file
-app.queue()  # Enable queueing for shared links
-app.launch(share=True, inline=False, custom_css="style.css")
+        
+app.launch(share=True)

styles.css

@@ -1,23 +0,0 @@
-body {
-    background-color: #121212;
-    color: #ffffff;
-}
-
-.gradio-container {
-    background-color: #121212;
-    color: #ffffff;
-}
-
-.gr-input, .gr-label, .gr-button, .gr-radio, .gr-slider {
-    background-color: #333333;
-    color: #ffffff;
-    border-color: #444444;
-}
-
-.gr-button {
-    background-color: #444444;
-}
-
-.gr-button:hover {
-    background-color: #555555;
-}