Update app.py

app.py

@@ -1,18 +1,18 @@
 import streamlit as st
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
-# Load the tokenizer and model
+st.set_page_config(page_title="GPT-2 Text Generator", layout="centered")
+
 tokenizer = AutoTokenizer.from_pretrained("ahmadmac/Pretrained-GPT2")
 model = AutoModelForCausalLM.from_pretrained("ahmadmac/Pretrained-GPT2")
 
-# Function to generate text
 def generate_text(prompt):
     inputs = tokenizer(prompt, return_tensors="pt")
-    outputs = model.generate(**inputs, max_length=150)
+    outputs = model.generate(**inputs, max_length=50)
     generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
     return generated_text
 
-st.title("GPT-2 Shakespear Text Generator")
+st.title("GPT-2 Text Generator")
 st.write("Enter a prompt to generate text using GPT-2")
 
 user_input = st.text_input("Prompt")
@@ -24,7 +24,3 @@ if st.button("Generate"):
             st.write(generated_text)
     else:
         st.warning("Please enter a prompt")
-
-if __name__ == "__main__":
-    st.set_page_config(page_title="GPT-2 Text Generator", layout="centered")
-    st.write("GPT-2 Text Generator")