Galuh Sahid commited on
Commit
35ae704
1 Parent(s): b4d2987

fix reload

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -106,9 +106,11 @@ elif model_name in ["GPT-2 Small finetuned on Indonesian academic journals"]:
106
  session_state = SessionState.get(prompt=None, prompt_box=None, text=None)
107
 
108
  ALL_PROMPTS = list(PROMPT_LIST[prompt_group_name].keys())+["Custom"]
109
- session_state.prompt = st.selectbox('Prompt', ALL_PROMPTS, index=len(ALL_PROMPTS)-1)
110
 
111
- if session_state.prompt is not None and (prompt != session_state.prompt):
 
 
112
  session_state.prompt_box = None
113
  session_state.text = None
114
 
 
106
  session_state = SessionState.get(prompt=None, prompt_box=None, text=None)
107
 
108
  ALL_PROMPTS = list(PROMPT_LIST[prompt_group_name].keys())+["Custom"]
109
+ prompt = st.selectbox('Prompt', ALL_PROMPTS, index=len(ALL_PROMPTS)-1)
110
 
111
+ if session_state.prompt is None:
112
+ session_state.prompt = prompt
113
+ elif session_state.prompt is not None and (prompt != session_state.prompt):
114
  session_state.prompt_box = None
115
  session_state.text = None
116