Now, pass your input to the model and return the logits:

logits = run_inference(trained_model, sample_test_video["video"])

Decoding the logits, we get: 
 

predicted_class_idx = logits.argmax(-1).item()
print("Predicted class:", model.config.id2label[predicted_class_idx])

Predicted class: BasketballDunk
```