Spaces:
Runtime error
Runtime error
print shape to make sure it works
Browse files
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']
|