fix reloading

app.py

@@ -14,7 +14,7 @@ LOGO = "huggingwayang.png"
 MODELS = {
     "GPT-2 Small": "flax-community/gpt2-small-indonesian",
     "GPT-2 Medium": "flax-community/gpt2-medium-indonesian",
-    "GPT-2 Small Finetuned on Indonesian Journals": "Galuh/id-journal-gpt2"
+    "GPT-2 Small finetuned on Indonesian academic journals": "Galuh/id-journal-gpt2"
 }
 
 headers = {}
@@ -96,11 +96,11 @@ st.markdown(
     """
 )
 
-model_name = st.selectbox('Model',(['GPT-2 Small', 'GPT-2 Medium', 'GPT-2 Small Finetuned on Indonesian Journals']))
+model_name = st.selectbox('Model',(['GPT-2 Small', 'GPT-2 Medium', 'GPT-2 Small finetuned on Indonesian academic journals']))
 
 if model_name in ["GPT-2 Small", "GPT-2 Medium"]:
     prompt_group_name = "GPT-2"
-elif model_name in ["GPT-2 Small Finetuned on Indonesian Journals"]:
+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"]
@@ -118,6 +118,8 @@ session_state.prompt_box = prompt_box
 text = st.text_area("Enter text", session_state.prompt_box)
 
 if st.button("Run"):
+    text = st.text_area("Enter text", session_state.prompt_box)
+
     with st.spinner(text="Getting results..."):
         lang_predictions, lang_probability = ft_model.predict(text.replace("\n", " "), k=3)
         if "__label__id" in lang_predictions: