HuggingfaceClient.Inference.Task.TextGeneration (huggingface_client v0.1.0)

Copy Markdown View Source

Text generation (completion) task.

Example

{:ok, result} = HuggingfaceClient.text_generation(client, %{
  model: "gpt2",
  inputs: "The answer to the universe is"
})
IO.puts(result["generated_text"])

Summary

Functions

run(client, args)
stream(client, args)

Functions

run(client, args)

@spec run(HuggingfaceClient.Client.t(), map()) ::
  {:ok, map()} | {:error, Exception.t()}

stream(client, args)

@spec stream(HuggingfaceClient.Client.t(), map()) ::
  {:ok, Enumerable.t()} | {:error, Exception.t()}