import gradio as gr
from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained("gpt2")
model = AutoModelForCausalLM.from_pretrained("gpt2")

def generate(text):
    inputs = tokenizer.encode(text, return_tensors='pt')
    outputs = model.generate(inputs, max_length=100, do_sample=True)
    generated = tokenizer.decode(outputs[0], skip_special_tokens=True)
    return generated

gr.Interface(fn=generate, inputs="text", outputs="text").launch()