Update handler.py

handler.py

@@ -18,7 +18,7 @@ class EndpointHandler:
             'donghuna/timesformer-base-finetuned-k400-diving48',
             ignore_mismatched_sizes=True
         )
-        self.model.classifier = torch.nn.Linear(self.model.classifier.in_features, 48)  # 48 output classes
+        # 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]) -> List[Dict[str, Any]]: