fix app

app.py

@@ -140,6 +140,7 @@ def infer(prompt):
     grid = torch.stack(frame_grids, dim=0)  # stack in temporal dim [t, 3, n*h, w]
     grid = (grid + 1.0) / 2.0
     grid = (grid * 255).to(torch.uint8).permute(0, 2, 3, 1)
+    print(grid.shape)
     torchvision.io.write_video(
         video_path,
         grid,