Inference

yaml
type: "io.kestra.plugin.huggingface.Inference"

Examples

yaml
id: huggingface_inference_text
namespace: company.team

tasks:
- id: huggingface_inference
  type: io.kestra.plugin.huggingface.Inference
  model: cardiffnlp/twitter-roberta-base-sentiment-latest
  apiKey: "{{ secret('HUGGINGFACE_API_KEY') }}"
  inputs: "I want a refund"

yaml
id: huggingface_inference
namespace: company.team

tasks:
- id: huggingface_inference_image
  type: io.kestra.plugin.huggingface.Inference
  model: google/vit-base-patch16-224
  apiKey: "{{ secret('HUGGINGFACE_API_KEY') }}"
  inputs: "{{ read('my-base64-image.txt') }}"
  parameters:
    function_to_apply: sigmoid,
    top_k: 3
  waitForModel: true
  useCache: false

Properties

apiKey *string

inputs *string

model *string

endpoint string

Defaulthttps://api-inference.huggingface.co/models

options

Definitions

io.kestra.plugin.huggingface.AbstractHttpTask-RequestOptions

connectTimeoutstring

connectionPoolIdleTimeoutstring

DefaultPT0S

defaultCharsetstring

DefaultUTF-8

java.nio.charset.Charset

maxContentLengthintegerstring

Default10485760

readIdleTimeoutstring

DefaultPT5M

readTimeoutstring

DefaultPT10S

parameters object

useCache booleanstring

Defaulttrue

waitForModel booleanstring

Defaultfalse

Outputs

output object

Use HuggingFace Inference API to classify customer message

Run DeepSeek chat and JSON completions

Tasks that call DeepSeek chat models for conversational or schema-guided JSON responses.

Perplexity

Tasks that request chat completions from Perplexity models.

Google Cloud

Execute tasks and orchestrate workflows across Google Cloud Platform services. Centralize authentication and resource management for seamless GCP integration.

CloudDataInfrastructureAI

Inference

Create automations with Huggingface Inference

Use HuggingFace Inference API to classify customer message

More Plugins in this Category

Run DeepSeek chat and JSON completions

Perplexity

Google Cloud

1.2.0