Update app.py

app.py

@@ -143,16 +143,25 @@ class SentenceBertJapanese:
 
 st.title("いらすと検索")
 description_text = st.empty()
-description_text.text("...モデル読み込み中...")
 
-model = SentenceBertJapanese("sonoisa/sentence-bert-base-ja-mean-tokens")
+if "model" not in st.session_state:
+    description_text.text("...モデル読み込み中...")
+    model = SentenceBertJapanese("sonoisa/sentence-bert-base-ja-mean-tokens")
+    st.session_state.model = model
 
-pyminizip.uncompress(
-    "irasuto_items_20210224.pq.zip", st.secrets["ZIP_PASSWORD"], None, 1
-)
-
-df = pq.read_table("irasuto_items_20210224.parquet").to_pandas()
-sentence_vectors = np.stack(df["sentence_vector"])
+    pyminizip.uncompress(
+        "irasuto_items_20210224.pq.zip", st.secrets["ZIP_PASSWORD"], None, 1
+    )
+    
+    df = pq.read_table("irasuto_items_20210224.parquet").to_pandas()
+    st.session_state.df = df
+    
+    sentence_vectors = np.stack(df["sentence_vector"])
+    st.session_state.sentence_vectors = sentence_vectors
+    
+model = st.session_state.model
+df = st.session_state.df
+sentence_vectors = st.session_state.sentence_vectors
 
 description_text.text("説明文の意味が近い「いらすとや」画像を検索します。\nキーワードを列挙するよりも、自然な文章を入力した方が精度よく検索できます。")