Update handler.py

ignore_mismatched_sizes=True

handler.py

@@ -10,7 +10,10 @@ import av
 
 class EndpointHandler:
     def __init__(self, model_dir=""):
-        self.model = TimesformerForVideoClassification.from_pretrained('donghuna/timesformer-base-finetuned-k400-diving48')
+        self.model = TimesformerForVideoClassification.from_pretrained(
+            '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.eval()