Aryanikale23 commited on
Commit
84f3939
1 Parent(s): ac3c4de

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -18,13 +18,13 @@ from torchvision.transforms import (
18
  RandomHorizontalFlip,
19
  Resize,
20
  )
21
- from transformers import VideoMAEFeatureExtractor, 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 = VideoMAEFeatureExtractor.from_pretrained(MODEL_CKPT)
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