tried fixing shapes

pages/1_Omniart_Disentanglement.py

@@ -170,7 +170,7 @@ else:
     original_image_vec = annotations['w_vectors'][st.session_state.image_id]
 
 if st.session_state.projection:
-    original_image_vec = original_image_vec - np.dot(original_image_vec.T, separation_vector) * separation_vector
+    original_image_vec = original_image_vec - np.dot(original_image_vec, separation_vector.T) * separation_vector
     print(original_image_vec.shape)
 
 img = generate_original_image(original_image_vec, model, latent_space=st.session_state.space_id)