chore: emmm

app.py

@@ -184,7 +184,7 @@ def generate_mesh(image, source_size=512, render_size=384, mesh_size=512, export
 
     with torch.no_grad():
         planes = model_wrapper.forward(image, source_camera)
-        planes_pil_image = ToPILImage()(planes[0][0])
+        planes_pil_image = ToPILImage()(planes[0])
 
         if export_mesh:
             grid_out = model_wrapper.model.synthesizer.forward_grid(planes=planes, grid_size=mesh_size)