Ahsen Khaliq
commited on
Commit
•
e1dddf5
1
Parent(s):
ee51e87
Update app.py
Browse files
app.py
CHANGED
@@ -5,17 +5,8 @@ import torch
|
|
5 |
import torch.nn.functional as F
|
6 |
import torchvision.transforms as T
|
7 |
from PIL import Image
|
8 |
-
from pytorchvideo.data.encoded_video import EncodedVideo
|
9 |
from torchvision.transforms._transforms_video import NormalizeVideo
|
10 |
|
11 |
-
from pytorchvideo.transforms import (
|
12 |
-
ApplyTransformToKey,
|
13 |
-
ShortSideScale,
|
14 |
-
UniformTemporalSubsample,
|
15 |
-
)
|
16 |
-
|
17 |
-
|
18 |
-
|
19 |
|
20 |
# Device on which to run the model
|
21 |
# Set to cuda to load on GPU
|
|
|
5 |
import torch.nn.functional as F
|
6 |
import torchvision.transforms as T
|
7 |
from PIL import Image
|
|
|
8 |
from torchvision.transforms._transforms_video import NormalizeVideo
|
9 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
10 |
|
11 |
# Device on which to run the model
|
12 |
# Set to cuda to load on GPU
|