Harishma01/Text_generation-task

1import gradio as gr2from transformers import pipeline3 4# Load GPT-2 text generation pipeline5generator = pipeline(6    "text-generation",7    model="gpt2"8)9 10def generate_text(prompt):11    if not prompt.strip():12        return "Please enter a prompt."13 14    result = generator(15        prompt,16        max_new_tokens=150,  # Adjust for approximately 15-20 lines17        temperature=0.7,18        top_k=50,19        top_p=0.95,20        do_sample=True,21        num_return_sequences=122    )23 24    return result[0]["generated_text"]25 26# Gradio Interface27demo = gr.Interface(28    fn=generate_text,29    inputs=gr.Textbox(30        lines=4,31        placeholder="Enter your prompt here..."32    ),33    outputs=gr.Textbox(34        lines=10,35        label="Generated Text"36    ),37    title="GPT-2 Text Generation",38    description="Generate meaningful text from a given prompt using a pre-trained GPT-2 model from Hugging Face Transformers."39)40 41demo.launch()