Update app.py

app.py

@@ -21,7 +21,7 @@ from PIL import Image
 from decord import VideoReader
 from decord import cpu
 from videomamba_image import videomamba_image_tiny
-from videomamba_video import videomamba_tiny
+from videomamba_video import videomamba_middle
 from kinetics_class_index import kinetics_classnames
 from imagenet_class_index import imagenet_classnames
 from transforms import (
@@ -39,7 +39,7 @@ device = "cuda"
 model_video_path = hf_hub_download(repo_id="OpenGVLab/VideoMamba", filename="videomamba_m16_k400_f16_res224.pth")
 model_image_path = hf_hub_download(repo_id="OpenGVLab/VideoMamba", filename="videomamba_t16_in1k_res224.pth")
 # Pick a pretrained model
-model_video = videomamba_tiny(num_classes=400, num_frames=16)
+model_video = videomamba_middle(num_classes=400, num_frames=16)
 video_sd = torch.load(model_video_path, map_location='cpu')
 model_video.load_state_dict(video_sd)
 model_image = videomamba_image_tiny()