JoJosmin commited on
Commit
c9990a6
·
verified ·
1 Parent(s): 394e02a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -183,7 +183,7 @@ elif st.session_state.step == 'show_results':
183
 
184
  # 세그먼트된 이미지에서 임베딩 추출
185
  #query_embedding = get_segmented_embedding(st.session_state.query_image, st.session_state.segmented_image)
186
- query_embedding = get_image_embedding(st.session_state.query_image, st.session_state.segmented_image)
187
  similar_images = find_similar_images(query_embedding, collection)
188
 
189
  st.subheader("Similar Items:")
 
183
 
184
  # 세그먼트된 이미지에서 임베딩 추출
185
  #query_embedding = get_segmented_embedding(st.session_state.query_image, st.session_state.segmented_image)
186
+ query_embedding = get_image_embedding(st.session_state.segmented_image)
187
  similar_images = find_similar_images(query_embedding, collection)
188
 
189
  st.subheader("Similar Items:")