File size: 1,141 Bytes
d95dc9f
 
 
 
 
 
 
 
 
 
 
 
 
 
867f29d
 
 
 
 
 
 
f6081d3
867f29d
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
---
tags:
- image-classification
library_name: generic
---

# Image Classification repository template

This is a template repository for image classification 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
* https://huggingface.co/osanseviero/fastai_cat_vs_dog/tree/main

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