Add set model

hupe1980 · Jul 7, 2023 · 900fe46 · 900fe46
1 parent 969b1f1
commit 900fe46
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/huggingface.go b/huggingface.go
@@ -57,6 +57,10 @@ func NewInferenceClient(token string, optFns ...func(o *InferenceClientOptions))
 	}
 }
 
+func (ic *InferenceClient) SetModel(model string) {
+	ic.opts.Model = model
+}
+
 // post sends a POST request to the specified model and task with the provided payload.
 // It returns the response body or an error if the request fails.
 func (ic *InferenceClient) post(ctx context.Context, model, task string, payload any) ([]byte, error) {