Spaces:
Runtime error
Runtime error
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']
|