Update handler.py

handler.py

@@ -32,9 +32,11 @@ class EndpointHandler:
 
         inputs = data.get("inputs")
         videos = read_video(inputs)
-        outputs = self.model(videos)
-        _, predicted = torch.max(outputs.logits, 1)
-        return predicted
+        with torch.no_grad():
+            outputs = self.model(videos)
+        logits = outputs.logits
+        _, predicted = torch.max(logits, 1)
+        return predicted.tolist()
         
         # inputs = data.get("inputs")
         # if not inputs: