Update app.py

app.py

@@ -1,10 +1,9 @@
-#pip install -r requirements.txt
 import streamlit as st
 from transformers import pipeline
 
-@st.cache(allow_output_mutation=True)
+@st.cache_resource(allow_output_mutation=True)
 def load_model():
-    return pipeline("text-generation", model="PeterBrendan/AdsGPT2")
+    return pipeline("text-generation", model="PeterBrendan/pbjs_gpt2")
 
 def main():
     st.title("Hugging Face Model Demo")
@@ -19,7 +18,7 @@ def main():
         generator = load_model()
 
         # Generate text based on user input
-        generated_text = generator(user_input, max_length=500, num_return_sequences=1)[0]["generated_text"]
+        generated_text = generator(user_input, max_length=100, num_return_sequences=1)[0]["generated_text"]
 
         # Display the generated text
         st.write("Generated Text:")