fix: device placement.

app.py

@@ -21,10 +21,11 @@ from torchvision.transforms import (
 from transformers import VideoMAEFeatureExtractor, VideoMAEForVideoClassification
 
 MODEL_CKPT = "sayakpaul/videomae-base-finetuned-kinetics-finetuned-ucf101-subset"
-MODEL = VideoMAEForVideoClassification.from_pretrained(MODEL_CKPT)
-PROCESSOR = VideoMAEFeatureExtractor.from_pretrained(MODEL_CKPT)
 DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
+MODEL = VideoMAEForVideoClassification.from_pretrained(MODEL_CKPT).to(DEVICE)
+PROCESSOR = VideoMAEFeatureExtractor.from_pretrained(MODEL_CKPT)
+
 RESIZE_TO = PROCESSOR.size
 NUM_FRAMES_TO_SAMPLE = MODEL.config.num_frames
 IMAGE_STATS = {"image_mean": [0.485, 0.456, 0.406], "image_std": [0.229, 0.224, 0.225]}