Change float32 to float64

_examples/feature_extraction/main.go

@@ -14,20 +14,30 @@ func main() {
 
 	res1, err1 := ic.FeatureExtraction(context.Background(), &huggingface.FeatureExtractionRequest{
 		Inputs: []string{"Hello World"},
+		Options: huggingface.Options{
+			WaitForModel: huggingface.PTR(true),
+		},
 	})
 	if err1 != nil {
 		log.Fatal(err1)
 	}
 
+	fmt.Println("FeatureExtraction:")
 	fmt.Println(res1[0])
+	fmt.Println()
 
 	res2, err2 := ic.FeatureExtractionWithAutomaticReduction(context.Background(), &huggingface.FeatureExtractionRequest{
 		Inputs: []string{"Hello World"},
 		Model:  "sentence-transformers/all-mpnet-base-v2",
+		Options: huggingface.Options{
+			WaitForModel: huggingface.PTR(true),
+		},
 	})
 	if err2 != nil {
 		log.Fatal(err2)
 	}
 
+	fmt.Println("FeatureExtractionWithAutomaticReduction:")
 	fmt.Println(res2[0])
+	fmt.Println()
 }

feature_extraction.go

@@ -15,10 +15,10 @@ type FeatureExtractionRequest struct {
 }
 
 // Response structure for the feature extraction endpoint
-type FeatureExtractionResponse [][][][]float32
+type FeatureExtractionResponse [][][][]float64
 
 // Response structure for the feature extraction endpoint
-type FeatureExtractionWithAutomaticReductionResponse [][]float32
+type FeatureExtractionWithAutomaticReductionResponse [][]float64
 
 // FeatureExtraction performs feature extraction using the specified model.
 // It sends a POST request to the Hugging Face inference endpoint with the provided input data.

huggingface.go

@@ -203,3 +203,7 @@ func contains[T comparable](collection []T, element T) bool {
 
 	return false
 }
+
+func PTR[T any](input T) *T {
+	return &input
+}