This is a template repository for image classification to support generic inference with Hugging Face Hub generic Inference API. There are two required steps
- Specify the requirements by defining a
- Implement the
__call__methods. These methods are called by the Inference API. The
__init__method should load the model and preload all the elements needed for inference (model, processors, tokenizers, etc.). This is only called once. The
__call__method performs the actual inference. Make sure to follow the same input/output specifications defined in the template for the pipeline to work.
First create a repo in https://hf.co/new. Then clone this template and push it to your repo.
git clone https://huggingface.co/templates/image-classification cd image-classification git remote set-url origin https://huggingface.co/$YOUR_USER/$YOUR_REPO_NAME git push --force
- Downloads last month