Spaces:
Sleeping
Sleeping
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:
|