handler

handler.py

@@ -13,11 +13,15 @@ class EndpointHandler:
         Args:
             data (:obj:): prediction input text
         """
-        # inputs = data.pop("inputs", data)
+        inputs = data.pop("inputs", data)
 
-        # start = perf_counter()
-        # prediction = self.pipeline(inputs)
-        # end = perf_counter()
-        # latency = end - start
+        start = perf_counter()
+        prediction = self.pipeline(inputs)
+        end = perf_counter()
+        latency = end - start
 
-        return {"greeting": "hi"}
+        return {
+            "labels": prediction.labels, 
+            "scores": prediction.scores,
+            "latency (secs.)": latency
+        }