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 *Requiredstring
inputs *Requiredstring
model *Requiredstring
endpoint string
Default
https://api-inference.huggingface.co/models
options AbstractHttpTask-RequestOptions
parameters object
useCache booleanstring
Default
true
waitForModel booleanstring
Default
false
Outputs
output object
Definitions
java.nio.charset.Charset
io.kestra.plugin.huggingface.AbstractHttpTask-RequestOptions
connectTimeout string
Format
duration
connectionPoolIdleTimeout string
Default
PT0S
Format
duration
defaultCharset Charsetstring
Default
UTF-8
maxContentLength integerstring
Default
10485760
readIdleTimeout string
Default
PT5M
Format
duration
readTimeout string
Default
PT10S
Format
duration