fix reload

app.py

@@ -106,7 +106,7 @@ elif model_name in ["GPT-2 Small finetuned on Indonesian academic journals"]:
 session_state = SessionState.get(prompt=None, prompt_box=None, text=None)
 
 ALL_PROMPTS = list(PROMPT_LIST[prompt_group_name].keys())+["Custom"]
-prompt = st.selectbox('Prompt', ALL_PROMPTS, index=len(ALL_PROMPTS)-1)
+session_state.prompt = st.selectbox('Prompt', ALL_PROMPTS, index=len(ALL_PROMPTS)-1)
 
 if session_state.prompt is not None and (prompt != session_state.prompt):
     session_state.prompt_box = None
@@ -115,7 +115,7 @@ if session_state.prompt is not None and (prompt != session_state.prompt):
 if session_state.prompt == "Custom":
     session_state.prompt_box = "Enter your text here"
 else:
-    if session_state.prompt_box is None:
+    if session_state.prompt is not None and session_state.prompt_box is None:
         session_state.prompt_box = random.choice(PROMPT_LIST[prompt_group_name][session_state.prompt])
 
 session_state.text = st.text_area("Enter text", session_state.prompt_box)