Update app.py

app.py

@@ -35,7 +35,7 @@ if st.button("Predict", type='primary'):
     response = requests.post("https://rda9527-deployment-test-v2.hf.space/v1/customer", json=customer_data)
     if response.status_code == 200:
         result = response.json()
-        churn_prediction = result["Churn expected?"]  # Extract only the value
+        churn_prediction = result["Prediction"]  # Extract only the value
         st.write(f"Based on the information provided, the customer with ID {CustomerID} is likely to {churn_prediction}.")
     else:
         st.error("Error in API request")