Update app.py

app.py

@@ -6,16 +6,19 @@ def load_model():
     return pipeline("text-generation", model="PeterBrendan/pbjs_gpt2")
 
 def main():
-    st.title("Pbjs GPT2: Optimize your Prebid config")
+    if "generated_widget_id" not in st.session_state:
+        st.session_state["generated_widget_id"] = None
+
+    st.title("Pbjs GPT2")
     st.write("Enter some text like **bidderTimeout** and get a generated Prebid config output.")
 
     # Default prompts
-    default_prompts = ["{", "bidderTimeout", "bidderSequence", "Usebidcache", "customPriceBucket"]
+    default_prompts = ["prompt 1", "prompt 2", "prompt 3"]
 
-    # Create a select box for default prompts
+    # Create a selectbox for default prompts
     default_prompt = st.selectbox("Choose a default prompt:", default_prompts)
 
-    # Create a text input field for a custom prompt
+    # Create a text input field for custom prompt
     custom_prompt = st.text_input("Enter a custom prompt:", "")
 
     # Check if a default prompt is selected