Spaces:
Running
on
Zero
Running
on
Zero
chore: planes has 1 item?
Browse files
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:
|