Upload folder using huggingface_hub

app.py

@@ -21,9 +21,9 @@ SeniorCitizen = st.selectbox("Is the customer a senior citizen?", ["Yes", "No"])
 
 # Convert categorical inputs to match model training
 customer_data = {
-    'Partner': 1 if Partner=="Yes" else 0,
-    'Dependents': 1 if Dependents=="Yes" else 0,
-    'PhoneService': 1 if PhoneService=="Yes" else 0,
+    'Partner': Partner,
+    'Dependents': Dependents,
+    'PhoneService': PhoneService,
     'InternetService': InternetService,
     'Contract': Contract,
     'PaymentMethod': PaymentMethod,
@@ -31,6 +31,7 @@ customer_data = {
     'MonthlyCharges': MonthlyCharges,
     'TotalCharges': TotalCharges,
     'SeniorCitizen':1 if SeniorCitizen == "Yes" else 0
+
 }
 
 if st.button("Predict", type='primary'):
@@ -38,7 +39,7 @@ if st.button("Predict", type='primary'):
     if response.status_code == 200:
         result = response.json()
         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}.")
+        st.write(f"Based on the information provided is likely to {churn_prediction}.")
     else:
         st.error("Error in API request")