From b83fb56664c774d6f88d4614a137ac51a53e7cd2 Mon Sep 17 00:00:00 2001 From: Thomas van Dongen Date: Tue, 11 Feb 2025 14:24:31 +0100 Subject: [PATCH] docs: Update README.md (#184) --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 8f02c0e..24a9142 100644 --- a/README.md +++ b/README.md @@ -103,7 +103,7 @@ from datasets import load_dataset from model2vec.train import StaticModelForClassification # Initialize a classifier from a pre-trained model -classifier = StaticModelForClassification.from_pretrained("minishlab/potion-base-8M") +classifier = StaticModelForClassification.from_pretrained(model_name="minishlab/potion-base-8M") # Load a dataset ds = load_dataset("setfit/subj")