fix app

app.py

@@ -127,7 +127,7 @@ def infer(prompt):
         x_T_total=x_T_total,
     )
 
-    video_path = "/tmp/output.mp4"
+    video_path = "output.mp4"
     vid_tensor = batch_samples[0]
     video = vid_tensor.detach().cpu()
     video = torch.clamp(video.float(), -1.0, 1.0)
@@ -140,7 +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,