Update app.py

app.py

@@ -1,10 +1,6 @@
-from transformers import GPT2LMHeadModel, GPT2Tokenizer
+from transformers import pipeline
+generator = pipeline("text-generation", model="gpt2")
+prompt = "Once upon a time, in a galaxy far, far away"
+generated_text = generator(prompt, max_length=100, num_return_sequences=3)
 
-model_id = "gpt2"
-model = GPT2LMHeadModel.from_pretrained(model_id)
-tokenizer = GPT2Tokenizer.from_pretrained(model_id)
-
-text = "Hello"
-encoded_text = tokenizer(text, return_tensors="pt") 
-
-outputs = model(**encoded_text)
+print(generated_text)