fix reloading

app.py

@@ -103,26 +103,26 @@ if model_name in ["GPT-2 Small", "GPT-2 Medium"]:
 elif model_name in ["GPT-2 Small finetuned on Indonesian academic journals"]:
     prompt_group_name = "Indonesian Journals"
 
-ALL_PROMPTS = list(PROMPT_LIST[prompt_group_name].keys())+["Custom"]
-prompt = st.selectbox('Prompt', ALL_PROMPTS, index=len(ALL_PROMPTS)-1)
+session_state = SessionState.get(prompt=None, prompt_box=None, text=None)
 
-session_state = SessionState.get(prompt_box=None)
+ALL_PROMPTS = list(PROMPT_LIST[prompt_group_name].keys())+["Custom"]
+session_state.prompt = st.selectbox('Prompt', ALL_PROMPTS, index=len(ALL_PROMPTS)-1)
 
-if prompt == "Custom":
+if session_state.prompt == "Custom":
     session_state.prompt_box = "Enter your text here"
 else:
-    session_state.prompt_box = random.choice(PROMPT_LIST[prompt_group_name][prompt])
+    session_state.prompt_box = random.choice(PROMPT_LIST[prompt_group_name][session_state.prompt])
 
-text = st.text_area("Enter text", session_state.prompt_box)
+session_state.text = st.text_area("Enter text", session_state.prompt_box)
 
 if st.button("Run"):
     with st.spinner(text="Getting results..."):
-        lang_predictions, lang_probability = ft_model.predict(text.replace("\n", " "), k=3)
+        lang_predictions, lang_probability = ft_model.predict(session_state.text.replace("\n", " "), k=3)
         if "__label__id" in lang_predictions:
             lang = "id"
         else:
             lang = lang_predictions[0].replace("__label__", "")
-            text = translate(text, "id", lang)
+            text = translate(session_state.text, "id", lang)
 
         st.subheader("Result")
         model = load_gpt(model_name)