Update handler.py

handler.py

@@ -24,20 +24,12 @@ class EndpointHandler:
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
         """
         data args:
-            inputs (:obj: `str`): base64 encoded video data
-            date (:obj: `str`)
+            inputs (:obj:`str`): base64 encoded video data
         Return:
-            A :obj:`list` | `dict`: will be serialized and returned
+            A :obj:`list` | `dict`: A list of dictionaries with the top 3 class indices and their probabilities
+                                   for each input video.
         """
 
-        # inputs = data.get("inputs")
-        # videos = read_video(inputs)
-        # with torch.no_grad():
-        #     outputs = self.model(videos)
-        # logits = outputs.logits
-        # _, predicted = torch.max(logits, 1)
-        # return predicted.tolist()
-
         inputs = data.get("inputs")
         videos = read_video(inputs)