Update handler.py (#1)

- Update handler.py (e01039cabaad6e7a64667038b465bcb649b8be46)


Co-authored-by: Philipp Schmid <philschmid@users.noreply.huggingface.co>

handler.py

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