By: Daniel Nazarian 🐧👹
Contact me at dnaz@danielnazarian.com
This repository contains a simple playground for zero-shot classification using the Hugging Face Transformers library. The goal is to provide a simple and easy-to-understand example of how to use the zero-shot classification pipeline provided by the Hugging Face Transformers library.
Some files included in this repository:
basic.py- The most basic example of using the zero-shot classification pipeline. Hardcoded image and labels.main.py- A more advanced example of using the zero-shot classification pipeline. Allows for user input of image and labels as well as methods for label handling.test.py- A simple test script to test different label method performance on the zero-shot classification pipeline.
Blog post coming soon 👀
Copyright © 2024 Daniel Nazarian