Update handler.py

handler.py

@@ -31,8 +31,10 @@ class EndpointHandler:
         """
 
         inputs = data.get("inputs")
-        read_video(inputs)
-        return inputs[0:10]
+        videos = read_video(inputs)
+        outputs = model(videos)
+        _, predicted = torch.max(outputs.logits, 1)
+        return predicted
         
         # inputs = data.get("inputs")
         # if not inputs: