metadata
tags:
- feature-extraction
library_name: generic
Feature Extraction repository template
This is a template repository for feature extraction to support generic inference with Hugging Face Hub generic Inference API. There are two required steps
- Specify the requirements by defining a
requirements.txtfile. - Implement the
pipeline.py__init__and__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.
Example repos