ludusc commited on
Commit
2422d19
1 Parent(s): 82608be

print shape to make sure it works

Browse files
Files changed (1) hide show
  1. pages/1_Omniart_Disentanglement.py +1 -0
pages/1_Omniart_Disentanglement.py CHANGED
@@ -171,6 +171,7 @@ else:
171
 
172
  if st.session_state.projection:
173
  original_image_vec = original_image_vec - np.dot(original_image_vec.T, separation_vector) * separation_vector
 
174
 
175
  img = generate_original_image(original_image_vec, model, latent_space=st.session_state.space_id)
176
  # input_image = original_image_dict['image']
 
171
 
172
  if st.session_state.projection:
173
  original_image_vec = original_image_vec - np.dot(original_image_vec.T, separation_vector) * separation_vector
174
+ print(original_image_vec.shape)
175
 
176
  img = generate_original_image(original_image_vec, model, latent_space=st.session_state.space_id)
177
  # input_image = original_image_dict['image']