Update app.py

app.py

@@ -55,8 +55,9 @@ def interpolate_vectors(vec1, vec2, t,importances=None):
     """
    # print(vec1.shape,vec2.shape)
     # Reshape to (511, 256)
-    vec1 = vec1.view(511, 256)
-    vec2 = vec2.view(511, 256)
+    print(vec1.shape)
+    vec1 = vec1.view(510, 256)
+    vec2 = vec2.view(510, 256)
     
     # Apply FFT along the first dimension
     fft1 = torch.fft.fft(vec1, dim=0)
@@ -91,7 +92,7 @@ def interpolate_vectors(vec1, vec2, t,importances=None):
     interpolated_seq = torch.fft.ifft(interpolated_fft, dim=0).real
     
     # Reshape back to (512, 1, 256)
-    interpolated_vec = interpolated_seq.view(511, 1,256)
+    interpolated_vec = interpolated_seq.view(510, 1,256)
     
     return interpolated_vec