github.com/hupe1980/go-huggingface@v0.0.15/_examples/feature_extraction/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  	res1, err1 := ic.FeatureExtraction(context.Background(), &huggingface.FeatureExtractionRequest{
    16  		Inputs: []string{"Hello World"},
    17  		Options: huggingface.Options{
    18  			WaitForModel: huggingface.PTR(true),
    19  		},
    20  	})
    21  	if err1 != nil {
    22  		log.Fatal(err1)
    23  	}
    24  
    25  	fmt.Println("FeatureExtraction:")
    26  	fmt.Println(res1[0])
    27  	fmt.Println()
    28  
    29  	res2, err2 := ic.FeatureExtractionWithAutomaticReduction(context.Background(), &huggingface.FeatureExtractionRequest{
    30  		Inputs: []string{"Hello World"},
    31  		Model:  "sentence-transformers/all-mpnet-base-v2",
    32  		Options: huggingface.Options{
    33  			WaitForModel: huggingface.PTR(true),
    34  		},
    35  	})
    36  	if err2 != nil {
    37  		log.Fatal(err2)
    38  	}
    39  
    40  	fmt.Println("FeatureExtractionWithAutomaticReduction:")
    41  	fmt.Println(res2[0])
    42  	fmt.Println()
    43  }