rajistics
/

churn-model

Tabular Classification

Inference Endpoints

Model card Files Files and versions Community

Update handler.py

#1

by philschmid HF staff - opened Oct 18, 2022

base: refs/heads/main

←

from: refs/pr/1

Discussion Files changed

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -22,7 +22,7 @@ class EndpointHandler():
         """
         inputs = data.pop("inputs", data)
         parameters = data.pop("parameters", None)
-        df = pd.DataFrame(data)
         df["TotalCharges"] = df["TotalCharges"].replace(" ", np.nan, regex=False).astype(float)
         df = df.drop(columns=["customerID"])

         """
         inputs = data.pop("inputs", data)
         parameters = data.pop("parameters", None)
+        df = pd.DataFrame(inputs)
         df["TotalCharges"] = df["TotalCharges"].replace(" ", np.nan, regex=False).astype(float)
         df = df.drop(columns=["customerID"])