Update app.py
Browse files
app.py
CHANGED
@@ -38,7 +38,7 @@ device = "cuda"
|
|
38 |
model_video_path = hf_hub_download(repo_id="OpenGVLab/VideoMamba", filename="videomamba_t16_k400_f32_res224.pth")
|
39 |
model_image_path = hf_hub_download(repo_id="OpenGVLab/VideoMamba", filename="videomamba_t16_in1k_res224.pth")
|
40 |
# Pick a pretrained model
|
41 |
-
model_video = videomamba_tiny(num_classes=400, num_frames=
|
42 |
video_sd = torch.load(model_video_path, map_location='cpu')
|
43 |
model_video.load_state_dict(video_sd)
|
44 |
model_image = videomamba_image_tiny()
|
|
|
38 |
model_video_path = hf_hub_download(repo_id="OpenGVLab/VideoMamba", filename="videomamba_t16_k400_f32_res224.pth")
|
39 |
model_image_path = hf_hub_download(repo_id="OpenGVLab/VideoMamba", filename="videomamba_t16_in1k_res224.pth")
|
40 |
# Pick a pretrained model
|
41 |
+
model_video = videomamba_tiny(num_classes=400, num_frames=32)
|
42 |
video_sd = torch.load(model_video_path, map_location='cpu')
|
43 |
model_video.load_state_dict(video_sd)
|
44 |
model_image = videomamba_image_tiny()
|