Random Sentence Generation with Huggingface Transformers

  • Share this:

Code introduction


This function uses the Huggingface Transformers library's text-generation pipeline to generate a random sentence based on the provided prompt.


Technology Stack : Huggingface Transformers

Code Type : The type of code

Code Difficulty : Intermediate


                
                    
def generate_random_sentence(prompt, max_length=50):
    from transformers import pipeline

    # Initialize a pipeline for text generation using a random model from Huggingface
    generator = pipeline('text-generation', model='gpt2')

    # Generate a random sentence based on the provided prompt
    generated_text = generator(prompt, max_length=max_length, num_return_sequences=1)[0]['generated_text']

    return generated_text.strip()