Random Text Generation with Huggingface Transformers

This function uses the text-generation pipeline from the Huggingface Transformers library to generate random text based on a given prompt.

Technology Stack : Huggingface Transformers, text-generation pipeline

Code Type : Custom function

Code Difficulty : Intermediate

                
                    
def generate_random_text(prompt, max_length=50, model="gpt2"):
    from transformers import pipeline

    # Load a text generation pipeline
    generator = pipeline("text-generation", model=model)

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