sayakpaul/ucf101-subset
Viewer • Updated • 2 • 850 • 2
How to use IsraelSonseca/videomae-base-finetuned-ucf101-subset with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("video-classification", model="IsraelSonseca/videomae-base-finetuned-ucf101-subset") # Load model directly
from transformers import AutoImageProcessor, AutoModelForVideoClassification
processor = AutoImageProcessor.from_pretrained("IsraelSonseca/videomae-base-finetuned-ucf101-subset")
model = AutoModelForVideoClassification.from_pretrained("IsraelSonseca/videomae-base-finetuned-ucf101-subset", device_map="auto")