Spaces:
Runtime error
Runtime error
Commit
•
84f3939
1
Parent(s):
ac3c4de
Update app.py
Browse files
app.py
CHANGED
@@ -18,13 +18,13 @@ from torchvision.transforms import (
|
|
18 |
RandomHorizontalFlip,
|
19 |
Resize,
|
20 |
)
|
21 |
-
from transformers import
|
22 |
|
23 |
MODEL_CKPT = "Aryanikale23/Signlanguage"
|
24 |
DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
25 |
|
26 |
MODEL = VideoMAEForVideoClassification.from_pretrained(MODEL_CKPT).to(DEVICE)
|
27 |
-
PROCESSOR =
|
28 |
|
29 |
RESIZE_TO = PROCESSOR.size["shortest_edge"]
|
30 |
NUM_FRAMES_TO_SAMPLE = MODEL.config.num_frames
|
|
|
18 |
RandomHorizontalFlip,
|
19 |
Resize,
|
20 |
)
|
21 |
+
from transformers import VideoMAEImageProcessor, VideoMAEForVideoClassification
|
22 |
|
23 |
MODEL_CKPT = "Aryanikale23/Signlanguage"
|
24 |
DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
25 |
|
26 |
MODEL = VideoMAEForVideoClassification.from_pretrained(MODEL_CKPT).to(DEVICE)
|
27 |
+
PROCESSOR = VideoMAEImageProcessor.from_pretrained(MODEL_CKPT)
|
28 |
|
29 |
RESIZE_TO = PROCESSOR.size["shortest_edge"]
|
30 |
NUM_FRAMES_TO_SAMPLE = MODEL.config.num_frames
|