Spaces:
Running
on
A10G
Running
on
A10G
update app.py for GLB output
#2
by
AP123
- opened
app.py
CHANGED
@@ -388,7 +388,7 @@ def stage2_run(models, device, tmp_dir,
|
|
388 |
mesh.vertices[:, 0] = -mesh.vertices[:, 0]
|
389 |
mesh.faces = np.fliplr(mesh.faces)
|
390 |
# Export the mesh as .obj file with colors
|
391 |
-
mesh.export(mesh_path, file_type='
|
392 |
torch.cuda.empty_cache()
|
393 |
|
394 |
if not is_rerun:
|
|
|
388 |
mesh.vertices[:, 0] = -mesh.vertices[:, 0]
|
389 |
mesh.faces = np.fliplr(mesh.faces)
|
390 |
# Export the mesh as .obj file with colors
|
391 |
+
mesh.export(mesh_path.replace('.obj', '.glb'), file_type='glb', include_color=True)
|
392 |
torch.cuda.empty_cache()
|
393 |
|
394 |
if not is_rerun:
|