GPT-2 Text Generation with Prompt

This function uses the GPT-2 model from the Huggingface Transformers library to generate random text based on an input prompt.

Technology Stack : Huggingface Transformers, GPT-2

Code Type : Text generation

Code Difficulty : Intermediate

                
                    
def generate_random_text(prompt, max_length=50):
    from transformers import pipeline
    # Load a pre-trained text generation pipeline
    generator = pipeline("text-generation", model="gpt2")

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