Pinwheel/ActsOfAgression
Viewer • Updated • 2k • 23 • 5
How to use archit11/videomae-base-finetuned-fight-nofight with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("video-classification", model="archit11/videomae-base-finetuned-fight-nofight") # Load model directly
from transformers import AutoImageProcessor, AutoModelForVideoClassification
processor = AutoImageProcessor.from_pretrained("archit11/videomae-base-finetuned-fight-nofight")
model = AutoModelForVideoClassification.from_pretrained("archit11/videomae-base-finetuned-fight-nofight", device_map="auto")This is experimentational if youre expecting this to work (it wont) or be useful should probably look eslewhere😛