yanranxiaoxi commited on
Commit
3ddce45
·
verified ·
1 Parent(s): cb249aa

chore: planes has 1 item?

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -185,8 +185,8 @@ def generate_mesh(image, source_size=512, render_size=384, mesh_size=512, export
185
  with torch.no_grad():
186
  planes = model_wrapper.forward(image, source_camera)
187
  planes_pil_image = np.concatenate([
188
- np.concatenate([planes[1], planes[2]], axis=1),
189
- np.concatenate([planes[3], planes[0]], axis=1),
190
  ], axis=0)
191
 
192
  if export_mesh:
 
185
  with torch.no_grad():
186
  planes = model_wrapper.forward(image, source_camera)
187
  planes_pil_image = np.concatenate([
188
+ np.concatenate([planes[0][1], planes[0][2]], axis=1),
189
+ np.concatenate([planes[0][3], planes[0][0]], axis=1),
190
  ], axis=0)
191
 
192
  if export_mesh: