Update README.md
Browse files
README.md
CHANGED
@@ -93,8 +93,8 @@ container = av.open(file_path)
|
|
93 |
indices = sample_frame_indices(clip_len=16, frame_sample_rate=1, seg_len=container.streams.video[0].frames)
|
94 |
video = read_video_pyav(container, indices)
|
95 |
|
96 |
-
image_processor = AutoImageProcessor.from_pretrained("videomae-base-finetuned-ucfcrime-full")
|
97 |
-
model = VideoMAEForVideoClassification.from_pretrained("videomae-base-finetuned-ucfcrime-full")
|
98 |
|
99 |
inputs = image_processor(list(video), return_tensors="pt")
|
100 |
|
|
|
93 |
indices = sample_frame_indices(clip_len=16, frame_sample_rate=1, seg_len=container.streams.video[0].frames)
|
94 |
video = read_video_pyav(container, indices)
|
95 |
|
96 |
+
image_processor = AutoImageProcessor.from_pretrained("archit11/videomae-base-finetuned-ucfcrime-full")
|
97 |
+
model = VideoMAEForVideoClassification.from_pretrained("archit11/videomae-base-finetuned-ucfcrime-full")
|
98 |
|
99 |
inputs = image_processor(list(video), return_tensors="pt")
|
100 |
|