Update app.py

app.py

@@ -182,8 +182,8 @@ elif st.session_state.step == 'show_results':
     st.image(original_image, caption="Original Image", use_column_width=True)
     
     # 세그먼트된 이미지에서 임베딩 추출
-    #query_embedding = get_segmented_embedding(st.session_state.query_image, st.session_state.segmented_image)
-    query_embedding = get_image_embedding(st.session_state.segmented_image)
+    query_embedding = get_segmented_embedding(st.session_state.query_image, st.session_state.segmented_image)
+    #query_embedding = get_image_embedding(st.session_state.segmented_image)
     similar_images = find_similar_images(query_embedding, collection)
     
     st.subheader("Similar Items:")