Create app.py

app.py

@@ -0,0 +1,19 @@
+from transformers import GPT2LMHeadModel, GPT2Tokenizer
+
+# Load the pre-trained GPT-2 model and tokenizer
+model_name = "gpt2"  # You can use other GPT-2 variants like "gpt2-medium" or "gpt2-large"
+model = GPT2LMHeadModel.from_pretrained(model_name)
+tokenizer = GPT2Tokenizer.from_pretrained(model_name)
+
+# Input text
+input_text = "Once upon a time, in a land far, far away..."
+
+# Tokenize the input text
+input_ids = tokenizer.encode(input_text, return_tensors="pt")
+
+# Generate text
+output = model.generate(input_ids, max_length=100, num_return_sequences=1)
+
+# Decode and print the generated text
+generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
+print(generated_text)