Update app.py

app.py

@@ -60,13 +60,9 @@ def read_video(video) -> torch.Tensor:
     """
     Reads a video file and converts it into a torch.Tensor with the shape [F, C, H, W].
     """
-    video_tensor = []
     to_tensor_transform = transforms.ToTensor()
     if isinstance(video, str):
-        with imageio.get_reader(video) as reader:
-            # Read all frames
-            for frame in reader:
-                video_tensor.append(to_tensor_transform(frame))
+        video_tensor = torch.stack([to_tensor_transform(img) for img in imageio.get_reader(video)])
     else: # video is a list of pil images
         video_tensor = torch.stack([to_tensor_transform(img) for img in video])
     return video_tensor
@@ -215,6 +211,7 @@ def generate_video(
         
         # Convert to tensor
         processed_video = read_video(processed_video)
+        print(type(processed_video))
         
         progress(0.2, desc="Preparing generation parameters...")