From 9c5541fc2e13a052ca8b182b30ae09f81bca1c40 Mon Sep 17 00:00:00 2001
From: hupe1980 <frankhuebner1980@gmail.com>
Date: Sun, 25 Jun 2023 19:18:54 +0200
Subject: [PATCH] Allow custom http client

---
 huggingface.go | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/huggingface.go b/huggingface.go
index ad06b85..969de84 100644
--- a/huggingface.go
+++ b/huggingface.go
@@ -25,6 +25,7 @@ type InferenceClientOptions struct {
 	Model             string
 	Endpoint          string
 	InferenceEndpoint string
+	HTTPClient        HTTPClient
 }
 
 // InferenceClient is a client for performing inference using Hugging Face models.
@@ -45,8 +46,12 @@ func NewInferenceClient(token string, optFns ...func(o *InferenceClientOptions))
 		fn(&opts)
 	}
 
+	if opts.HTTPClient == nil {
+		opts.HTTPClient = http.DefaultClient
+	}
+
 	return &InferenceClient{
-		httpClient: http.DefaultClient,
+		httpClient: opts.HTTPClient,
 		token:      token,
 		opts:       opts,
 	}