archit11 commited on
Commit
d3dda12
1 Parent(s): d2a2733

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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