Update app.py

app.py

@@ -14,10 +14,10 @@ def generate(slider_x, slider_y, prompt, x_concept_1, x_concept_2, y_concept_1,
 
     # check if avg diff for directions need to be re-calculated
     if not sorted(slider_x) == sorted([x_concept_1, x_concept_2]):
-        clip_slider.avg_diff = clip_slider.find_latent_direction(slider_x[0], slider_x[0])
+        clip_slider.avg_diff = clip_slider.find_latent_direction(slider_x[0], slider_x[1])
         x_concept_1, x_concept_2 = slider_x[0], slider_x[1]
     if not sorted(slider_y) == sorted([y_concept_1, y_concept_2]):
-        clip_slider.avg_diff_2nd = clip_slider.find_latent_direction(slider_y[0], slider_y[0])
+        clip_slider.avg_diff_2nd = clip_slider.find_latent_direction(slider_y[0], slider_y[1])
         y_concept_1, y_concept_2 = slider_y[0], slider_y[1]
     
     comma_concepts_x = ', '.join(slider_x)