Update app.py

app.py

@@ -31,15 +31,15 @@ def generate(slider_x, slider_y, prompt, seed, iterations, steps,
     print("x_concept_1", x_concept_1, "x_concept_2", x_concept_2)
     if not sorted(slider_x) == sorted([x_concept_1, x_concept_2]):
         avg_diff = clip_slider.find_latent_direction(slider_x[0], slider_x[1], num_iterations=iterations)
-        avg_diff[0].to(torch.float16)
-        avg_diff[1].to(torch.float16)
+        avg_diff[0] = avg_diff[0].to(torch.float16)
+        avg_diff[1] = avg_diff[1].to(torch.float16)
         x_concept_1, x_concept_2 = slider_x[0], slider_x[1]
     
     print("avg_diff[0].dtype", avg_diff[0].dtype)
     if not sorted(slider_y) == sorted([y_concept_1, y_concept_2]):
         avg_diff_2nd = clip_slider.find_latent_direction(slider_y[0], slider_y[1], num_iterations=iterations)
-        avg_diff_2nd[0].to(torch.float16)
-        avg_diff_2nd[1].to(torch.float16)
+        avg_diff_2nd[0] = avg_diff_2nd[0].to(torch.float16)
+        avg_diff_2nd[1] = avg_diff_2nd[1].to(torch.float16)
         y_concept_1, y_concept_2 = slider_y[0], slider_y[1]
     end_time = time.time()
     print(f"direction time: {end_time - start_time:.2f} ms")