Fix app.py

app.py

@@ -310,30 +310,32 @@ def encode_image(image_filename, model):
     return image_embedding
 
 
-st.title("いらすと検索（日本語CLIP ゼロショット）")
+st.title("いらすと検索（日本語CLIPゼロショット）")
 description_text = st.empty()
 
 if "model" not in st.session_state:
-    description_text.text("...日本語CLIPモデル読み込み中...")
+    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
 
+    print("extract dataset")
     pyminizip.uncompress(
         "clip_zeroshot_irasuto_items_20210224.pq.zip", st.secrets["ZIP_PASSWORD"], None, 1
     )
     
+    print("loading dataset")
     df = pq.read_table("clip_zeroshot_irasuto_items_20210224.parquet").to_pandas()
     st.session_state.df = df
     
-    sentence_vectors = np.stack(df["sentence_vector"])
+    # sentence_vectors = np.stack(df["sentence_vector"])
     image_vectors = np.stack(df["image_vector"])
-    st.session_state.sentence_vectors = sentence_vectors
+    # st.session_state.sentence_vectors = sentence_vectors
     st.session_state.image_vectors = image_vectors
     
 model = st.session_state.model
 df = st.session_state.df
-sentence_vectors = st.session_state.sentence_vectors
+# sentence_vectors = st.session_state.sentence_vectors
 image_vectors = st.session_state.image_vectors
 
 description_text.text("日本語CLIPモデル（ゼロショット）を用いて、説明文の意味が近い「いらすとや」画像を検索します。\nキーワードを列挙するよりも、自然な文章を入力した方が精度よく検索できます。\n画像は必ずリンク先の「いらすとや」さんのページを開き、そこからダウンロードしてください。")