diff --git a/sentence_transformers/model_card.py b/sentence_transformers/model_card.py index 99da35d96..83eb11623 100644 --- a/sentence_transformers/model_card.py +++ b/sentence_transformers/model_card.py @@ -423,7 +423,7 @@ def set_widget_examples(self, dataset: Dataset | DatasetDict) -> None: columns = [ column for column, feature in dataset[dataset_name].features.items() - if isinstance(feature, Value) and feature.dtype == "string" and column != "dataset_name" + if isinstance(feature, Value) and (feature.dtype == "string" or feature.dtype == "large_string") and column != "dataset_name" ] str_dataset = dataset[dataset_name].select_columns(columns) dataset_size = len(str_dataset)