Update handler.py

handler.py

@@ -7,17 +7,12 @@ class EndpointHandler:
         self.model = joblib.load(path + "/valid_trial_v3_2.joblib")
 
     def __call__(self, data: Dict[str, Any]) -> List[Any]:
-        """
-         data args:
-              inputs (:obj: `str`)
-              date (:obj: `str`)
-        Return:
-              A :obj:`list`
-        """
+
         # get inputs
-        input = data.pop("input", data)
-        input_series = pd.Series(input)
+        s_in = data.pop("inputs", data)
+        s_in = pd.Series(s_in)
+        d_in = pd.DataFrame(s_in).transpose()
 
         # run normal prediction
-        prediction = self.model.predict_proba(input_series)[:, 1]
+        prediction = self.model.predict_proba(d_in)[:, 1]
         return prediction