Update app.py

app.py

@@ -1,5 +1,15 @@
 import streamlit as st
-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)
+from transformers import GPT2Tokenizer, GPT2Model
+
+tokenizer = GPT2Tokenizer.from_pretrained('gpt2')
+model = GPT2Model.from_pretrained('gpt2')
+
+st.title("GPT-2 Text Generation")
+
+text = st.text_input("Enter the starting text:", "Replace me by any text you'd like.")
+
+if st.button("Generate"):
+    encoded_input = tokenizer(text, return_tensors='pt')
+    output = model(**encoded_input)
+    generated_text = tokenizer.decode(output[0][0][encoded_input[0][0].tolist().index(tokenizer.eos_token_id):])
+    st.write(generated_text)