Update app.py

app.py

@@ -62,11 +62,7 @@ def generate_and_export_animation(prompt):
             processed_frames = []
             for frame_tensor in output.frames:
                 # Convert frame tensor to numpy array
-                frame_numpy = frame_tensor.cpu().numpy()
-            
-                # # Ensure frame has the correct shape (H, W, C)
-                # if len(frame_numpy.shape) == 4:
-                frame_numpy = frame_numpy.squeeze(0)
+                frame_numpy = frame_tensor.numpy()
             
                 # Append processed frame to the list
                 processed_frames.append(frame_numpy)