Hugging Face's logo

templates
/
feature-extraction

Feature Extraction generic
Model card Files Community
feature-extraction / README.md
osanseviero's picture
osanseviero HF staff
Update README.md
f3beae7
preview code
|
raw history blame contribute delete
No virus
1.12 kB
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

  1. Specify the requirements by defining a requirements.txt file.
  2. 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

How to start

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/feature-extraction
cd feature-extraction
git remote set-url origin https://huggingface.co/$YOUR_USER/$YOUR_REPO_NAME
git push --force