Update README.md

README.md

@@ -1,8 +1,9 @@
 ---
-license: "cc-by-nc-4.0"
+license: cc-by-nc-4.0
 tags:
 - vision
 - video-classification
+pipeline_tag: video-classification
 ---
 
 # VideoMAE-v2 (base-sized model, Pretrained on UnlabeledHybrid-1M) 
@@ -24,13 +25,15 @@ import numpy as np
 import torch
 
 
-config = AutoConfig.from_pretrained("./", trust_remote_code=True)
-model = AutoModel.from_pretrained('./', config=config, trust_remote_code=True)
+config = AutoConfig.from_pretrained("OpenGVLab/VideoMAEv2-Base", trust_remote_code=True)
+processor = VideoMAEImageProcessor.from_pretrained("OpenGVLab/VideoMAEv2-Base")
+model = AutoModel.from_pretrained('OpenGVLab/VideoMAEv2-Base', config=config, trust_remote_code=True)
+
 
 video = list(np.random.rand(16, 3, 224, 224))
 
-processor = VideoMAEImageProcessor.from_pretrained("./")
-model = AutoModel.from_pretrained("./",config=config, trust_remote_code=True)
+
+
 
 # B, T, C, H, W -> B, C, T, H, W
 inputs = processor(video, return_tensors="pt")