Update app.py

app.py

@@ -58,7 +58,22 @@ def generate_and_export_animation(prompt):
         with tempfile.NamedTemporaryFile(suffix='.mp4', delete=False) as temp_file:
             temp_video_path = temp_file.name
             print('temp_video_path', temp_video_path)
-            export_to_video(output.frames, temp_video_path)
+
+            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)
+            
+                # Append processed frame to the list
+                processed_frames.append(frame_numpy)
+
+            # Export processed frames to a video file
+            export_to_video(processed_frames, temp_video_path)
+            # export_to_video(output.frames, temp_video_path)
 
         with open(temp_video_path, 'rb') as video_file:
             video_binary = video_file.read()