Import Models from Hugging Face

The Hugging Face model definition allows you to import models from a Hugging Face repository.

#Feature

Currently, VDP supports importing models tagged with the following tasks

✅ Image Classification

#Release stage

Alpha

#Configuration

Field	Type	Note
`repo_id`*	string	A model repo name hosted on the Hugging Face model hub. Valid repo ids can be located at the root-level, or namespaced under a user or organization name, like `google/vit-base-patch16-224`

#Getting started

#Requirements

A public Hugging Face model repository that is labelled with Image Classification task

#No-code setup

To import a model from Hugging Face in the Console, do the following:

Go to the Model page and click Add new model
In the Set Up New Model page, fill an ID for your model, this will be the unique identifier of this model
[Optional] Give a short description of your model in the Description field
Click the Model source ▾ drop-down and choose Hugging Face
Fill the Hugging Face model ID that stores the model files and click Set up
Now go to the Model page, the corresponding model should be here. Note that it may take some time for the model to be deployed online.

#Low-code setup

Send a HTTP request to the VDP model-backend to import a model from Hugging Face.

cURL

curl -X POST http://localhost:8080/v1alpha/models -d '{
  "id": "vit-base-patch16-224",
  "model_definition": "model-definitions/huggingface",
  "configuration": {
    "repo_id": "google/vit-base-patch16-224"
  }
}'

Deploy the imported model vit-base-patch16-224.

cURL

curl -X POST http://localhost:8080/v1alpha/models/vit-base-patch16-224/deploy

Perform an inference to test the model

cURL(url)

cURL(base64)

cURL(multipart)

curl -X POST http://localhost:8080/v1alpha/models/vit-base-patch16-224/test -d '{
  "inputs": [
    {
      "image_url": "https://artifacts.instill.tech/imgs/dog.jpg"
    }
  ]
}'