Add radio button to select target vectors

app.py

@@ -330,9 +330,9 @@ if "model" not in st.session_state:
     
     sentence_vectors = np.stack(df["sentence_vector"])
     image_vectors = np.stack(df["image_vector"])
-    st.session_state.sentence_vectors = sentence_vectors
 
     st.session_state.df = df
+    st.session_state.sentence_vectors = sentence_vectors
     st.session_state.image_vectors = image_vectors
 
     print("finished loading model and dataset")
@@ -352,7 +352,7 @@ query_input = st.text_input(label="説明文", value="", on_change=clear_result)
 
 closest_n = st.number_input(label="検索数", min_value=1, value=10, max_value=100) 
 
-model_type = st.radio(label="検索対象ベクトル", options=("文", "画像"))
+model_type = st.radio(label="検索対象ベクトル", options=("画像", "文"))
 
 search_buttion = st.button("検索")
 
@@ -368,7 +368,7 @@ if search_buttion or prev_query != query_input:
         target_vectors = sentence_vectors
 
     distances = scipy.spatial.distance.cdist(
-        query_embedding, image_vectors, metric="cosine"
+        query_embedding, target_vectors, metric="cosine"
     )[0]
 
     results = zip(range(len(distances)), distances)