github.com/hupe1980/go-huggingface@v0.0.15/_examples/fill_mask/main.go (about)

     1  package main
     2  
     3  import (
     4  	"context"
     5  	"fmt"
     6  	"log"
     7  	"os"
     8  
     9  	"github.com/hupe1980/go-huggingface"
    10  )
    11  
    12  func main() {
    13  	ic := huggingface.NewInferenceClient(os.Getenv("HUGGINGFACEHUB_API_TOKEN"))
    14  
    15  	res, err := ic.FillMask(context.Background(), &huggingface.FillMaskRequest{
    16  		Inputs: []string{"The answer to the universe is <mask>."},
    17  	})
    18  	if err != nil {
    19  		log.Fatal(err)
    20  	}
    21  
    22  	for _, r := range res {
    23  		fmt.Println("Sequence:", r.Sequence)
    24  		fmt.Println("Score:", r.Score)
    25  		fmt.Println("TokenID:", r.TokenID)
    26  		fmt.Println("TokenStr", r.TokenStr)
    27  		fmt.Println("---")
    28  	}
    29  }