Computer Vision, AI, Transformers, Machine Learning, Open Source.
name_or_path=name_or_path, classes=classes ) You can then apply the model for image classification just as you did in the traditional image classification setting with apply_model(). Zero-Shot Object Detection works the same way, but with model type “zero-shot-detection-transformer-torch”: import