Update app.py
Browse files
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"
|