Update app.py

app.py

@@ -119,21 +119,27 @@ def postprocess_title(title):
 st.title("Qiita記事タイトル案生成")
 
 description_text = st.empty()
-description_text.text("...モデル読み込み中...")
 
-trained_model, tokenizer = load_model()
+if "trained_model" not in st.session_state:
+    description_text.text("...モデル読み込み中...")
+
+    trained_model, tokenizer = load_model()
+    trained_model.eval()
+    
+    st.session_state.trained_model = trained_model
+    st.session_state.tokenizer = tokenizer
+
+trained_model = st.session_state.trained_model
+tokenizer = st.session_state.tokenizer
 
 # GPUの利用有無
 USE_GPU = torch.cuda.is_available()
-# 推論モード設定
-trained_model.eval()
 
 description_text.text("記事の本文をコピペ入力して、タイトル生成ボタンを押すと、タイトル案が10個生成されます。\nGPUが使えないため生成に数十秒かかります。")
-body_text = st.text_area(label="記事の本文", value="", height=300, max_chars=4000)
+qiita_body = st.text_area(label="記事の本文", value="", height=300, max_chars=4000)
 answer = st.button("タイトル生成")
 
 if answer:
-    qiita_body = str(body_text)
     title_fieids = st.empty()
     title_fieids.markdown("...生成中...")