Update app.py

app.py

@@ -190,7 +190,7 @@ def generate_mesh(image, source_size=512, render_size=384, mesh_size=384, export
                 frames.append(frame_chunk['images_rgb'])
 
             frames = torch.cat(frames, dim=1)
-            print(frames.shape)
+            frames = frames.squeeze(0)
             frames = (frames.permute(0, 2, 3, 1).cpu().numpy() * 255).astype(np.uint8)
 
             video_path = "awesome_video.mp4"