Update app.py

app.py

@@ -65,11 +65,10 @@ st.markdown(
     """Thai GPT-2 demo. Part of the [Huggingface JAX/Flax event](https://discuss.huggingface.co/t/open-to-the-community-community-week-using-jax-flax-for-nlp-cv/)."""
 )
 model_name = st.selectbox('Model', (['GPT-2 Base']))
-prompt = st.selectbox('Prompt')
+hello = ['Custom', 'Yellow']
+prompt = st.selectbox('Prompt', options = hello)
 if prompt == "Custom":
     prompt_box = "Enter your text here"
-else:
-    prompt_box = random.choice(PROMPT_LIST[prompt])
 text = st.text_area("Enter text", prompt_box)
 if st.button("Run"):
     with st.spinner(text="Getting results..."):