ludusc commited on
Commit
c63daa2
1 Parent(s): 3b262b0
Files changed (1) hide show
  1. pages/1_Omniart_Disentanglement.py +1 -1
pages/1_Omniart_Disentanglement.py CHANGED
@@ -168,7 +168,7 @@ else:
168
  original_image_vec = annotations['w_vectors'][st.session_state.image_id]
169
 
170
  if projection:
171
- original_image_vec = original_image_vec - np.dot(original_image_vec, separation_vector) * separation_vector
172
 
173
  img = generate_original_image(original_image_vec, model, latent_space=st.session_state.space_id)
174
  # input_image = original_image_dict['image']
 
168
  original_image_vec = annotations['w_vectors'][st.session_state.image_id]
169
 
170
  if projection:
171
+ original_image_vec = original_image_vec - np.dot(original_image_vec.T, separation_vector) * separation_vector
172
 
173
  img = generate_original_image(original_image_vec, model, latent_space=st.session_state.space_id)
174
  # input_image = original_image_dict['image']