Shawon16 commited on
Commit
50f72af
·
verified ·
1 Parent(s): e20213e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -0
app.py CHANGED
@@ -13,6 +13,16 @@ from pytorchvideo.transforms import (
13
  )
14
  from torchvision.transforms import Compose, Lambda, Resize
15
  from transformers import VideoMAEFeatureExtractor, VideoMAEForVideoClassification
 
 
 
 
 
 
 
 
 
 
16
 
17
  # Load model and processor
18
  MODEL_CKPT = "Shawon16/VideoMAE_BdSLW401_20_epochs_p5_SR_10"
 
13
  )
14
  from torchvision.transforms import Compose, Lambda, Resize
15
  from transformers import VideoMAEFeatureExtractor, VideoMAEForVideoClassification
16
+ # FIXED IMPORT
17
+ from torchvision.transforms import functional as F
18
+
19
+ # --- Fix pytorchvideo import error for Kaggle/torchvision >= 0.17 ---
20
+ import sys
21
+ import types
22
+
23
+ # Create a fake module to satisfy pytorchvideo
24
+ fake_ft = types.ModuleType("torchvision.transforms.functional_tensor")
25
+ sys.modules["torchvision.transforms.functional_tensor"] = fake_ft
26
 
27
  # Load model and processor
28
  MODEL_CKPT = "Shawon16/VideoMAE_BdSLW401_20_epochs_p5_SR_10"