Add image query

app.py

@@ -310,11 +310,14 @@ def encode_image(image_filename, model):
     return image_embedding
 
 
-@st.cache
-def load_model_and_data():
+st.title("いらすと検索（日本語CLIPゼロショット）")
+description_text = st.empty()
+
+if "model" not in st.session_state:
     description_text.text("日本語CLIPモデル読み込み中... ")
     device = "cuda" if torch.cuda.is_available() else "cpu"
     model = ClipModel("sonoisa/clip-vit-b-32-japanese-v1", device=device)
+    st.session_state.model = model
 
     pyminizip.uncompress(
         "clip_zeroshot_irasuto_items_20210224.pq.zip", st.secrets["ZIP_PASSWORD"], None, 1
@@ -326,13 +329,15 @@ def load_model_and_data():
     sentence_vectors = np.stack(df["sentence_vector"])
     image_vectors = np.stack(df["image_vector"])
 
-    return model, df, sentence_vectors, image_vectors
+    st.session_state.df = df
+    st.session_state.sentence_vectors = sentence_vectors
+    st.session_state.image_vectors = image_vectors
 
 
-st.title("いらすと検索（日本語CLIPゼロショット）")
-description_text = st.empty()
-
-model, df, sentence_vectors, image_vectors = load_model_and_data()
+model = st.session_state.model
+df = st.session_state.df
+sentence_vectors = st.session_state.sentence_vectors
+image_vectors = st.session_state.image_vectors
 
 description_text.text("日本語CLIPモデル（ゼロショット）を用いて、説明文の意味が近い「いらすとや」画像を検索します。\nキーワードを列挙するよりも、自然な文章を入力した方が精度よく検索できます。\n画像は必ずリンク先の「いらすとや」さんのページを開き、そこからダウンロードしてください。")