Update app.py

app.py

@@ -36,10 +36,10 @@ from huggingface_hub import hf_hub_download
 # Device on which to run the model
 # Set to cuda to load on GPU
 device = "cuda"
-model_video_path = hf_hub_download(repo_id="OpenGVLab/VideoMamba", filename="videomamba_t16_k400_f32_res224.pth")
+model_video_path = hf_hub_download(repo_id="OpenGVLab/VideoMamba", filename="videomamba_t16_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=32)
+model_video = videomamba_tiny(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()