import gradio as gr
from transformers import pipeline

model_name = "EleutherAI/gpt-neo-1.3B"

def generate_text(prompt, max_length):
    generator = pipeline('text-generation', model=model_name, use_auth_token=None)
    result = generator(prompt, max_length=int(max_length), num_return_sequences=1)
    return result[0]['generated_text']

with gr.Blocks() as demo:
    gr.Markdown("# AI Text Generator")
    gr.Markdown("Powered by EleutherAI/gpt-neo-1.3B")
    prompt_input = gr.Textbox(label="Prompt", placeholder="Enter your prompt here...")
    max_len_input = gr.Slider(minimum=10, maximum=200, value=50, label="Max Length")
    output = gr.Textbox(label="Generated Text")
    btn = gr.Button("Generate")
    btn.click(fn=generate_text, inputs=[prompt_input, max_len_input], outputs=output)

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