diff --git a/src/transformers/pipelines/text_to_audio.py b/src/transformers/pipelines/text_to_audio.py index 5676aa6c21..f2c7e1dd4d 100644 --- a/src/transformers/pipelines/text_to_audio.py +++ b/src/transformers/pipelines/text_to_audio.py @@ -34,8 +34,8 @@ class TextToAudioPipeline(Pipeline): ```python >>> from transformers import pipeline - >>> classifier = pipeline(model="suno/bark") - >>> output = pipeline("Hey it's HuggingFace on the phone!") + >>> pipe = pipeline(model="suno/bark-small") + >>> output = pipe("Hey it's HuggingFace on the phone!") >>> audio = output["audio"] >>> sampling_rate = output["sampling_rate"] diff --git a/utils/documentation_tests.txt b/utils/documentation_tests.txt index 1fbbc4ee9b..0649f6e028 100644 --- a/utils/documentation_tests.txt +++ b/utils/documentation_tests.txt @@ -489,3 +489,4 @@ src/transformers/models/yolos/image_processing_yolos.py src/transformers/models/yolos/modeling_yolos.py src/transformers/models/yoso/configuration_yoso.py src/transformers/pipelines/ +src/transformers/pipelines/text_to_audio.py \ No newline at end of file