Update app.py

app.py

@@ -8,7 +8,7 @@ import torchvision.transforms as T
 from PIL import Image
 from decord import VideoReader
 from decord import cpu
-from uniformerv2 import uniformerv2_b16
+from uniformerv2 import uniformerv2_b16,uniformerv2_l14_336
 from kinetics_class_index import kinetics_classnames
 from transforms import (
     GroupNormalize, GroupScale, GroupCenterCrop, 
@@ -31,7 +31,7 @@ class Uniformerv2(nn.Module):
 device = "cpu"
 model_path = hf_hub_download(repo_id="lierzhao/mit_uniformerv2_l14_8x336.pyth", filename="mit_uniformerv2_l14_8x336.pyth")
 # Pick a pretrained model 
-model = Uniformerv2(uniformerv2_b16(pretrained=False, t_size=8, no_lmhra=True, temporal_downsample=False))
+model = Uniformerv2(uniformerv2_l14_336(pretrained=False, t_size=8, no_lmhra=True, temporal_downsample=False))
 state_dict = torch.load(model_path, map_location='cpu')
 model.load_state_dict(state_dict)