Update handler.py

handler.py

@@ -21,7 +21,9 @@ class EndpointHandler:
         self.model.classifier = torch.nn.Linear(self.model.classifier.in_features, 48)  # 48 output classes
         self.model.eval()
         
-    def __call__(self, data: Dict[str, Any]) -> Dict[str, Any]:
+    def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
+        return {"predicted_class": 1}
+        
         inputs = data.get("inputs")
         if not inputs:
             return {"error": "No video input provided"}