Update README.md

README.md

@@ -5,3 +5,9 @@ from transformers import pipeline, set_seed
 generator = pipeline('text-generation', model='gpt2')
 set_seed(42)
 generator("Hello, I'm a language model,", max_length=30, num_return_sequences=5)
+
+input_text = "The future of AI is"
+inputs = tokenizer(input_text, return_tensors="pt")
+
+output = model.generate(**inputs, max_length=100)
+print(tokenizer.decode(output[0], skip_special_tokens=True))