from transformers import pipeline, set_seed
import gradio as gr

generator = pipeline('text-generation', model='Keyurjotaniya007/openai-gpt2-wikitext-textgen')
set_seed(42)

def generate_text(prompt):
    response = generator(
        prompt,
        max_length=100,
        num_return_sequences=1,
        truncation=True
    )
    return response[0]['generated_text']

interface = gr.Interface(
    fn=generate_text,
    inputs=gr.Textbox(lines=5, placeholder="write something to generate..."),
    outputs="text",
    title="OpenAI GPT-2🦄",
    description="Run text generation with the model `Keyurjotaniya007/openai-gpt2-wikitext-textgen`, a GPT-2 variant fine-tuned for coherent long-form text."
)

if __name__ == "__main__":
    interface.launch()